robin-sparkless 4.0.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

"""
Tests for issue #250: Column.between(low, high) missing for range filter.

PySpark's Column has between(low, high) for inclusive range filters. Robin Column
now exposes between so df.filter(col("v").between(lit(20), lit(30))) works.
"""

from __future__ import annotations

from tests.fixtures.spark_imports import get_spark_imports

_imports = get_spark_imports()
F = _imports.F


def test_column_has_between(spark) -> None:
    """Column has between method (PySpark parity)."""
    c = F.col("v")
    assert hasattr(c, "between")


def test_filter_between_returns_matching_row(spark) -> None:
    """df.filter(col("v").between(lit(20), lit(30))) returns only the row where v is in [20, 30]."""
    data = [{"v": 10}, {"v": 25}, {"v": 50}]
    df = spark.createDataFrame(data, ["v"])
    out = df.filter(F.col("v").between(F.lit(20), F.lit(30))).collect()
    assert len(out) == 1
    assert out[0]["v"] == 25


def test_filter_between_inclusive_boundaries(spark) -> None:
    """between is inclusive: v between 10 and 10 returns the row with v=10."""
    data = [{"v": 10}, {"v": 20}]
    df = spark.createDataFrame(data, ["v"])
    out = df.filter(F.col("v").between(F.lit(10), F.lit(10))).collect()
    assert len(out) == 1
    assert out[0]["v"] == 10