test: div by zero returns nan

tests/expr_and_series/arithmetic_test.py

@@ -157,6 +157,101 @@ def test_truediv_same_dims(
     assert_equal_data({"a": result}, {"a": [2, 1, 1 / 3]})
 
 
+@pytest.mark.parametrize(
+    ("left", "right", "expected"),
+    [(-2, 0, float("-inf")), (0, 0, None), (2, 0, float("inf"))],
+)
+def test_series_truediv_by_zero(
+    left: int, right: int, expected: float | None, constructor_eager: ConstructorEager
+) -> None:
+    data: dict[str, list[int]] = {"a": [left], "b": [right]}
+    df = nw.from_native(constructor_eager(data), eager_only=True)
+    truediv_result = df["a"] / df["b"]
+    assert_equal_data({"a": truediv_result}, {"a": [expected]})
+
+
+@pytest.mark.parametrize(
+    ("left", "right", "expected"),
+    [(-2, 0, float("-inf")), (0, 0, None), (2, 0, float("inf"))],
+)
+@pytest.mark.skipif(PANDAS_VERSION < (2, 0), reason="converts floordiv by zero to 0")
+def test_series_floordiv_by_zero(
+    left: int,
+    right: int,
+    expected: float | None,
+    constructor_eager: ConstructorEager,
+    request: pytest.FixtureRequest,
+) -> None:
+    data: dict[str, list[int]] = {"a": [left], "b": [right]}
+    df = nw.from_native(constructor_eager(data), eager_only=True)
+    # pyarrow backend floordiv raises divide by zero error
+    if "pyarrow" in str(constructor_eager):
+        request.applymarker(pytest.mark.xfail)
+    # polars backend floordiv by zero always returns null
+    if "polars" in str(constructor_eager):
+        floordiv_result = df["a"] // df["b"]
+        assert all(floordiv_result.is_null())
+    # pandas[nullable] backend floordiv always returns 0
+    elif all(x in str(constructor_eager) for x in ["pandas", "nullable"]):
+        floordiv_result = df["a"] // df["b"]
+        assert_equal_data({"a": floordiv_result}, {"a": [0]})
+    else:
+        floordiv_result = df["a"] // df["b"]
+        assert_equal_data({"a": floordiv_result}, {"a": [expected]})
+
+
+@pytest.mark.parametrize(
+    ("left", "right", "expected"),
+    [(-2, 0, float("-inf")), (0, 0, None), (2, 0, float("inf"))],
+)
+def test_truediv_by_zero(
+    left: int, right: int, expected: float | None, constructor: Constructor
+) -> None:
+    data: dict[str, list[int]] = {"a": [left]}
+    df = nw.from_native(constructor(data))
+    truediv_result = df.select(nw.col("a") / right)
+    assert_equal_data(truediv_result, {"a": [expected]})
+
+
+@pytest.mark.parametrize(
+    ("left", "right", "expected"),
+    [(-2, 0, float("-inf")), (0, 0, None), (2, 0, float("inf"))],
+)
+@pytest.mark.skipif(PANDAS_VERSION < (2, 0), reason="converts floordiv by zero to 0")
+def test_floordiv_by_zero(
+    left: int,
+    right: int,
+    expected: float | None,
+    constructor: Constructor,
+    request: pytest.FixtureRequest,
+) -> None:
+    data: dict[str, list[int]] = {"a": [left]}
+    df = nw.from_native(constructor(data))
+    # pyarrow backend floordiv raises divide by zero error
+    # ibis backend floordiv cannot cast value to inf or -inf
+    if any(x in str(constructor) for x in ["ibis", "pyarrow"]):
+        request.applymarker(pytest.mark.xfail)
+    # duckdb backend floordiv return None
+    if "duckdb" in str(constructor):
+        floordiv_result = df.select(nw.col("a") // right)
+        assert_equal_data(floordiv_result, {"a": [None]})
+    # polars backend floordiv returns null
+    elif "polars" in str(constructor) and "lazy" not in str(constructor):
+        floordiv_result = df.select(nw.col("a") // right)
+        assert all(floordiv_result["a"].is_null())
+    # polars lazy floordiv cannot be sliced and returns None
+    elif all(x in str(constructor) for x in ["polars", "lazy"]):
+        floordiv_result = df.select(nw.col("a") // right)
+        assert_equal_data(floordiv_result, {"a": [None]})
+    # pandas[nullable] backend floordiv always returns 0
+    elif all(x in str(constructor) for x in ["pandas", "nullable"]):
+        floordiv_result = df.select(nw.col("a") // right)
+        assert_equal_data(floordiv_result, {"a": [0]})
+    else:
+        floordiv_result = df.select(nw.col("a") // right)
+        assert_equal_data(floordiv_result, {"a": [expected]})
+
+
 @pytest.mark.slow
 @given(left=st.integers(-100, 100), right=st.integers(-100, 100))
 @pytest.mark.skipif(PANDAS_VERSION < (2, 0), reason="convert_dtypes not available")