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

"""Regression test for #1412: array of struct built from literals inside select."""

from tests.fixtures.spark_imports import get_spark_imports


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

    df = spark.range(1, 2).select(
        F.array(F.struct(F.lit(1).alias("x"), F.lit("a").alias("s"))).alias("arr")
    )

    rows = df.collect()
    assert len(rows) == 1
    arr = rows[0]["arr"]
    assert len(arr) == 1
    assert arr[0]["x"] == 1
    assert arr[0]["s"] == "a"