diff --git a/tests/test_tables.py b/tests/test_tables.py index c61b3c72b..4a9078e9e 100644 --- a/tests/test_tables.py +++ b/tests/test_tables.py @@ -142,6 +142,17 @@ def test_column(table): assert_array_equal(t.column('letter'), np.array(['a', 'b', 'c', 'z'])) assert_array_equal(t.column(1), np.array([9, 3, 3, 1])) +def test_values(): + t1 = Table().with_columns({ + 'row1': ['a', 'b', 'c'], + 'row2': ['d', 'e', 'f'], + }) + assert_array_equal(t1.values, np.array(t1.columns, None).T) + t2 = Table().with_columns({ + 'row1': ['x', 'y', 'z'], + 'row2': [1, 2, 3], + }) + assert_array_equal(t2.values, np.array(t2.columns, object).T) def test_basic_points(table): t = table