ARROW-8285: [Python][Dataset] Test that ScalarExpression accepts nump…

…y scalars Add numpy scalar input to `test_expression_serialization()` in `test_dataset.py`. Closes #11545 from AlenkaF/ARROW-8285 Authored-by: Alenka Frim <[email protected]> Signed-off-by: Joris Van den Bossche <[email protected]>
apache · Oct 27, 2021 · a044fc6 · a044fc6
1 parent 1779e94
commit a044fc6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/pyarrow/tests/test_dataset.py b/python/pyarrow/tests/test_dataset.py
@@ -548,8 +548,9 @@ def test_expression_serialization():
     e = ds.scalar(None)
     f = ds.scalar({'a': 1})
     g = ds.scalar(pa.scalar(1))
+    h = ds.scalar(np.int64(2))
 
-    all_exprs = [a, b, c, d, e, f, g, a == b, a > b, a & b, a | b, ~c,
+    all_exprs = [a, b, c, d, e, f, g, h, a == b, a > b, a & b, a | b, ~c,
                  d.is_valid(), a.cast(pa.int32(), safe=False),
                  a.cast(pa.int32(), safe=False), a.isin([1, 2, 3]),
                  ds.field('i64') > 5, ds.field('i64') == 5,