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

"""Regression test for #1410: array.slice parity."""

from tests.fixtures.spark_imports import get_spark_imports


def test_array_slice_parity(spark):
    imports = get_spark_imports()
    F = imports.F

    df = spark.createDataFrame(
        [(["a", "b", "c"],), (["x"],), (None,)],
        ["arr"],
    )

    # Match PySpark: slice(arr, 2, 2) → ["b", "c"], [], None
    df2 = df.select(F.slice("arr", 2, 2).alias("sliced"))
    rows = df2.collect()

    assert len(rows) == 3
    assert rows[0]["sliced"] == ["b", "c"]
    assert rows[1]["sliced"] == []
    assert rows[2]["sliced"] is None