Merge pull request #604 from iwatat/add_test_table_apply

add test for iterable passed as columns argument for Table.apply
data-8 · Nov 29, 2023 · 2e0c20d · 2e0c20d
2 parents 2c706e2 + c53043e
commit 2e0c20d
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'),