def test_to_python(): a = Series("a", range(20)) b = a.to_list() assert isinstance(b, list) assert len(b) == 20 a = Series("a", [1, None, 2], nullable=True) assert a.null_count() == 1 assert a.to_list() == [1, None, 2]
def test_strategy_null_probability( s: pl.Series, df1: pl.DataFrame, df2: pl.DataFrame, df3: pl.DataFrame, ) -> None: for obj in (s, df1, df2, df3): assert len(obj) == 50 # type: ignore[arg-type] assert s.null_count() < df1.null_count().fold(sum).sum() assert df1.null_count().fold(sum).sum() < df2.null_count().fold(sum).sum() assert df2.null_count().fold(sum).sum() < df3.null_count().fold(sum).sum() nulls_col0, nulls_col1 = df2.null_count().rows()[0] assert nulls_col0 > nulls_col1 assert nulls_col0 < 50 nulls_col0, nulls_colx = df3.null_count().rows()[0] assert nulls_col0 > nulls_colx assert nulls_col0 == 50