Update tests

py-polars/tests/unit/namespaces/string/test_concat.py

@@ -0,0 +1,68 @@
+from datetime import datetime
+
+import polars as pl
+from polars.testing import assert_series_equal
+
+
+def test_str_concat() -> None:
+    s = pl.Series(["1", None, "2", None])
+    # propagate null
+    assert_series_equal(
+        s.str.concat("-", ignore_nulls=False), pl.Series([None], dtype=pl.String)
+    )
+    # ignore null
+    assert_series_equal(s.str.concat("-"), pl.Series(["1-2"]))
+
+    # str None/null is ok
+    s = pl.Series(["1", "None", "2", "null"])
+    assert_series_equal(
+        s.str.concat("-", ignore_nulls=False), pl.Series(["1-None-2-null"])
+    )
+    assert_series_equal(s.str.concat("-"), pl.Series(["1-None-2-null"]))
+
+
+def test_str_concat2() -> None:
+    df = pl.DataFrame({"foo": [1, None, 2, None]})
+
+    out = df.select(pl.col("foo").str.concat("-", ignore_nulls=False))
+    assert out.item() is None
+
+    out = df.select(pl.col("foo").str.concat("-"))
+    assert out.item() == "1-2"
+
+
+def test_str_concat_all_null() -> None:
+    s = pl.Series([None, None, None], dtype=pl.String)
+    assert_series_equal(
+        s.str.concat("-", ignore_nulls=False), pl.Series([None], dtype=pl.String)
+    )
+    assert_series_equal(s.str.concat("-", ignore_nulls=True), pl.Series([""]))
+
+
+def test_str_concat_empty_list() -> None:
+    s = pl.Series([], dtype=pl.String)
+    assert_series_equal(s.str.concat("-", ignore_nulls=False), pl.Series([""]))
+    assert_series_equal(s.str.concat("-", ignore_nulls=True), pl.Series([""]))
+
+
+def test_str_concat_empty_list2() -> None:
+    s = pl.Series([], dtype=pl.String)
+    df = pl.DataFrame({"foo": s})
+    result = df.select(pl.col("foo").str.concat("-")).item()
+    expected = ""
+    assert result == expected
+
+
+def test_str_concat_empty_list_agg_context() -> None:
+    df = pl.DataFrame(data={"i": [1], "v": [None]}, schema_overrides={"v": pl.String})
+    result = df.group_by("i").agg(pl.col("v").drop_nulls().str.concat("-"))["v"].item()
+    expected = ""
+    assert result == expected
+
+
+def test_str_concat_datetime() -> None:
+    df = pl.DataFrame({"d": [datetime(2020, 1, 1), None, datetime(2022, 1, 1)]})
+    out = df.select(pl.col("d").str.concat("|", ignore_nulls=True))
+    assert out.item() == "2020-01-01 00:00:00.000000|2022-01-01 00:00:00.000000"
+    out = df.select(pl.col("d").str.concat("|", ignore_nulls=False))
+    assert out.item() is None

py-polars/tests/unit/namespaces/string/test_string.py

@@ -1,8 +1,5 @@
 from __future__ import annotations
 
-from datetime import datetime
-from typing import cast
-
 import pytest
 
 import polars as pl
@@ -16,70 +13,6 @@ def test_str_slice() -> None:
     assert df.select([pl.col("a").str.slice(2, 4)])["a"].to_list() == ["obar", "rfoo"]
 
 
-def test_str_concat() -> None:
-    s = pl.Series(["1", None, "2", None])
-    # propagate null
-    assert_series_equal(
-        s.str.concat(ignore_nulls=False), pl.Series([None], dtype=pl.String)
-    )
-    # ignore null
-    assert_series_equal(s.str.concat(), pl.Series(["1-2"]))
-
-    # str None/null is ok
-    s = pl.Series(["1", "None", "2", "null"])
-    assert_series_equal(s.str.concat(ignore_nulls=False), pl.Series(["1-None-2-null"]))
-    assert_series_equal(s.str.concat(), pl.Series(["1-None-2-null"]))
-
-
-def test_str_concat2() -> None:
-    df = pl.DataFrame({"foo": [1, None, 2, None]})
-
-    out = df.select(pl.col("foo").str.concat("-", ignore_nulls=False))
-    assert cast(str, out.item()) is None
-
-    out = df.select(pl.col("foo").str.concat("-"))
-    assert cast(str, out.item()) == "1-2"
-
-
-def test_str_concat_all_null() -> None:
-    s = pl.Series([None, None, None], dtype=pl.String)
-    assert_series_equal(
-        s.str.concat(ignore_nulls=False), pl.Series([None], dtype=pl.String)
-    )
-    assert_series_equal(s.str.concat(ignore_nulls=True), pl.Series([""]))
-
-
-def test_str_concat_empty_list() -> None:
-    s = pl.Series([], dtype=pl.String)
-    assert_series_equal(s.str.concat(ignore_nulls=False), pl.Series([""]))
-    assert_series_equal(s.str.concat(ignore_nulls=True), pl.Series([""]))
-
-
-def test_str_concat_empty_list2() -> None:
-    s = pl.Series([], dtype=pl.String)
-    df = pl.DataFrame({"foo": s})
-    result = df.select(pl.col("foo").str.concat()).item()
-    expected = ""
-    assert result == expected
-
-
-def test_str_concat_empty_list_agg_context() -> None:
-    df = pl.DataFrame(data={"i": [1], "v": [None]}, schema_overrides={"v": pl.String})
-    result = df.group_by("i").agg(pl.col("v").drop_nulls().str.concat())["v"].item()
-    expected = ""
-    assert result == expected
-
-
-def test_str_concat_datetime() -> None:
-    df = pl.DataFrame({"d": [datetime(2020, 1, 1), None, datetime(2022, 1, 1)]})
-    out = df.select(pl.col("d").str.concat("|", ignore_nulls=True))
-    assert (
-        cast(str, out.item()) == "2020-01-01 00:00:00.000000|2022-01-01 00:00:00.000000"
-    )
-    out = df.select(pl.col("d").str.concat("|", ignore_nulls=False))
-    assert cast(str, out.item()) is None
-
-
 def test_str_len_bytes() -> None:
     s = pl.Series(["Café", None, "345", "東京"])
     result = s.str.len_bytes()