robin-sparkless 0.11.5

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

"""
Tests for #354: Column name attribute (PySpark parity).

PySpark's Column has a .name attribute for simple column references (e.g. F.col("x").name → "x").
"""

from __future__ import annotations

import robin_sparkless as rs


def test_column_name_simple_reference() -> None:
    """col("salary").name returns "salary" (PySpark parity)."""
    c = rs.col("salary")
    assert c.name == "salary"


def test_column_name_other_names() -> None:
    """col("x").name and col("other_col").name return the given name."""
    assert rs.col("x").name == "x"
    assert rs.col("other_col").name == "other_col"


def test_column_name_aliased() -> None:
    """Aliased column .name returns the alias."""
    c = rs.col("salary").alias("sal")
    assert c.name == "sal"