From 0601f5ecb406c048989a3098d137ef94d3eb4d43 Mon Sep 17 00:00:00 2001 From: Itamar Turner-Trauring Date: Tue, 23 Jul 2024 16:41:53 -0400 Subject: [PATCH] Specify dtype explicitly --- .../unit/operations/arithmetic/test_arithmetic.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/py-polars/tests/unit/operations/arithmetic/test_arithmetic.py b/py-polars/tests/unit/operations/arithmetic/test_arithmetic.py index 33d85c2ab96b..ba6b00ffcdc4 100644 --- a/py-polars/tests/unit/operations/arithmetic/test_arithmetic.py +++ b/py-polars/tests/unit/operations/arithmetic/test_arithmetic.py @@ -562,11 +562,15 @@ def test_power_series() -> None: @pytest.mark.parametrize( ("expected", "expr", "column_names"), [ - (np.array([[2, 4], [6, 8]]), lambda a, b: a + b, ("a", "a")), - (np.array([[0, 0], [0, 0]]), lambda a, b: a - b, ("a", "a")), - (np.array([[1, 4], [9, 16]]), lambda a, b: a * b, ("a", "a")), - (np.array([[1.0, 1.0], [1.0, 1.0]]), lambda a, b: a / b, ("a", "a")), - (np.array([[0, 0], [0, 0]]), lambda a, b: a % b, ("a", "a")), + (np.array([[2, 4], [6, 8]], dtype=np.int64), lambda a, b: a + b, ("a", "a")), + (np.array([[0, 0], [0, 0]], dtype=np.int64), lambda a, b: a - b, ("a", "a")), + (np.array([[1, 4], [9, 16]], dtype=np.int64), lambda a, b: a * b, ("a", "a")), + ( + np.array([[1.0, 1.0], [1.0, 1.0]], dtype=np.float64), + lambda a, b: a / b, + ("a", "a"), + ), + (np.array([[0, 0], [0, 0]], dtype=np.int64), lambda a, b: a % b, ("a", "a")), ( np.array([[3, 4], [7, 8]], dtype=np.int64), lambda a, b: a + b,