test: specify dtypes

scikit-hep · Jan 31, 2023 · b3612f6 · b3612f6
1 parent ca41a8c
commit b3612f6
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tests/test_2020_reduce_axis_none.py b/tests/test_2020_reduce_axis_none.py
@@ -272,12 +272,14 @@ def test_all():
 
 
 def test_prod_numpy():
-    as_numpy = ak.from_numpy(np.arange(7 * 5 * 3).reshape((7, 5, 3)))
-    assert ak.sum(as_numpy, axis=None) == 5460
+    array_reg = ak.from_numpy(np.arange(7 * 5 * 3, dtype=np.int64).reshape((7, 5, 3)))
+    result_reg = ak.from_numpy(np.array([[[5460]]], dtype=np.int64))
+
+    assert ak.sum(array_reg, axis=None) == 5460
     assert ak._util.arrays_approx_equal(
-        ak.sum(as_numpy, axis=None, keepdims=True), np.array([[[5460]]], dtype=np.int64)
+        ak.sum(array_reg, axis=None, keepdims=True), result_reg
     )
     assert ak._util.arrays_approx_equal(
-        ak.sum(as_numpy, axis=None, keepdims=True, mask_identity=True),
-        ak.to_regular(ak.Array([[[5460]]]).mask[[[[True]]]], axis=None),
+        ak.sum(array_reg, axis=None, keepdims=True, mask_identity=True),
+        ak.to_regular(result_reg.mask[[[[True]]]], axis=None),
     )