set type

pola-rs · May 25, 2024 · 46545fa · 46545fa
1 parent 9544f67
commit 46545fa
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/py-polars/tests/unit/datatypes/test_array.py b/py-polars/tests/unit/datatypes/test_array.py
@@ -299,8 +299,7 @@ def test_recursive_array_dtype() -> None:
 
 
 def test_ndarray_construction() -> None:
-    a = np.arange(16).reshape((2, 4, -1))
+    a = np.arange(16, dtype=np.int64).reshape((2, 4, -1))
     s = pl.Series(a)
-    print(s.dtype)
     assert s.dtype == pl.Array(pl.Int64, (4, 2))
     assert (s.to_numpy() == a).all()