Merge pull request #488 from slim-builder/test_tables_col

Add ValueError test for tables.column method
data-8 · Feb 11, 2021 · cb81e9d · cb81e9d
2 parents 834cf6e + 65e9605
commit cb81e9d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_tables.py b/tests/test_tables.py
@@ -141,6 +141,10 @@ def test_column(table):
     t = 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]))
+    with pytest.raises(ValueError):
+        t.column(-1)
+    with pytest.raises(ValueError):
+        t.column('abc')
 
 def test_values():
     t1 = Table().with_columns({