add test for iterable passed as columns argument for Table.apply

data-8 · Nov 25, 2023 · c53043e · c53043e
1 parent 2c706e2
commit c53043e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_tables.py b/tests/test_tables.py
@@ -676,6 +676,9 @@ def test_apply(table):
                        np.array([18, 6, 6, 2]))
     with(pytest.raises(ValueError)):
         t.apply(lambda x, y: x + y, 'count', 'score')
+    with(pytest.warns(FutureWarning)):
+        assert_array_equal(t.apply(lambda x: x * x, ['points']),
+                           np.array([1, 4, 4, 100]))
 
     # Deprecated behavior
     assert_array_equal(t.apply(lambda x, y: x * y, 'count', 'points'),