robin-sparkless 4.4.0

PySpark-like DataFrame API in Rust on Polars; no JVM.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

from sparkless.testing import get_imports

_imports = get_imports()
F = _imports.F

"""Tests for issue #218: Division by zero returns null (PySpark parity).

PySpark returns null for division by zero; these tests assert that behavior
directly against a real PySpark session via the ``spark`` fixture.
"""


def test_division_by_zero_literal_over_column(spark) -> None:
    """Exact scenario from #218: lit(1) / col('x') with x=0 -> null."""
    df = spark.createDataFrame(
        [{"x": 1}, {"x": 0}],
        schema=["x"],
    )
    result = df.withColumn("q", F.lit(1) / F.col("x"))
    rows = result.collect()
    assert len(rows) == 2
    assert rows[0]["q"] == 1.0
    assert rows[1]["q"] is None


def test_division_by_zero_column_over_literal(spark) -> None:
    """col('x') / lit(0) -> null."""
    df = spark.createDataFrame(
        [{"x": 10}, {"x": 20}],
        schema=["x"],
    )
    result = df.withColumn("q", F.col("x") / F.lit(0))
    rows = result.collect()
    assert len(rows) == 2
    assert rows[0]["q"] is None
    assert rows[1]["q"] is None


def test_division_by_zero_column_over_column(spark) -> None:
    """col('a') / col('b') with b=0 -> null."""
    df = spark.createDataFrame(
        [{"a": 5, "b": 1}, {"a": 5, "b": 0}],
        schema=["a", "b"],
    )
    result = df.withColumn("q", F.col("a") / F.col("b"))
    rows = result.collect()
    assert rows[0]["q"] == 5.0
    assert rows[1]["q"] is None