fix bad kwarg

dask · Jun 15, 2021 · baf368a · baf368a
1 parent 86c5f19
commit baf368a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/dask/array/tests/test_ufunc.py b/dask/array/tests/test_ufunc.py
@@ -478,14 +478,14 @@ def test_divmod():
 
 @pytest.mark.parametrize("dt", ["float64", "float32", "int32", "int64"])
 def test_dtype_kwarg(dt):
-    x1 = da.asarray([1, 2, 3], dtype=np.int64)
-    y1 = da.asarray([4, 5, 6], dtype=np.int64)
+    x1 = da.asarray([1, 2, 3])
+    y1 = da.asarray([4, 5, 6])
     z1 = np.add(x1, y1, dtype=np.dtype(dt))
     assert z1.dtype == np.dtype(dt)
     assert z1.compute().dtype == z1.dtype
 
-    x2 = da.asarray([1, 2, 3], dtype=np.int64)
-    y2 = da.asarray([4, 5, 6], dtype=np.int64)
+    x2 = da.asarray([1, 2, 3])
+    y2 = da.asarray([4, 5, 6])
     z2 = da.add(x2, y2, dtype=np.dtype(dt))
     assert z2.dtype == np.dtype(dt)
     assert z2.compute().dtype == z2.dtype