def test_non_exact_strptime() -> None:
a = pl.Series(
"a", ["2022-01-16", "2022-01-17", "foo2022-01-18", "b2022-01-19ar"])
fmt = "%Y-%m-%d"
expected = pl.Series(
"a",
[date(2022, 1, 16), date(2022, 1, 17), None, None])
verify_series_and_expr_api(a,
expected,
"str.strptime",
pl.Date,
fmt,
strict=False,
exact=True)
expected = pl.Series(
"a",
[
date(2022, 1, 16),
date(2022, 1, 17),
date(2022, 1, 18),
date(2022, 1, 19)
],
)
verify_series_and_expr_api(a,
expected,
"str.strptime",
pl.Date,
fmt,
strict=False,
exact=False)
with pytest.raises(Exception):
a.str.strptime(pl.Date, fmt, strict=True, exact=True)
def test_list_join_strings() -> None:
s = pl.Series("a", [["ab", "c", "d"], ["e", "f"], ["g"], []])
expected = pl.Series("a", ["ab-c-d", "e-f", "g", ""])
verify_series_and_expr_api(s, expected, "arr.join", "-")
def test_unique_counts() -> None:
s = pl.Series("id", ["a", "b", "b", "c", "c", "c"])
expected = pl.Series("id", [1, 2, 3], dtype=pl.UInt32)
verify_series_and_expr_api(s, expected, "unique_counts")
def test_unique_stable() -> None:
a = pl.Series("a", [1, 1, 1, 1, 2, 2, 2, 3, 3])
expected = pl.Series("a", [1, 2, 3])
verify_series_and_expr_api(a, expected, "unique", True)
def test_list_argminmax() -> None:
s = pl.Series("a", [[1, 2], [3, 2, 1]])
expected = pl.Series("a", [0, 2], dtype=pl.UInt32)
verify_series_and_expr_api(s, expected, "arr.arg_min")
expected = pl.Series("a", [1, 0], dtype=pl.UInt32)
verify_series_and_expr_api(s, expected, "arr.arg_max")