feat(rust, python): add support for serializing categoricals to json (#…

…7276)
pola-rs · Mar 1, 2023 · 258a1b4 · 258a1b4
1 parent 21a3881
commit 258a1b4
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -42,7 +42,7 @@ package = "arrow2"
 git = "https://github.com/ritchie46/arrow2"
 # rev = "7be484eb530805696c1ab79aae809f973d92cd5a"
 # path = "../arrow2"
-branch = "polars_2023-02-22"
+branch = "polars_2023-03-01"
 version = "0.16"
 default-features = false
 features = [

diff --git a/py-polars/Cargo.lock b/py-polars/Cargo.lock
diff --git a/py-polars/tests/unit/io/test_json.py b/py-polars/tests/unit/io/test_json.py
@@ -103,3 +103,13 @@ def test_ndjson_nested_utf8_int() -> None:
     assert pl.read_ndjson(io.StringIO(ndjson)).to_dict(False) == {
         "Accumulables": [[{"Value": "32395888"}, {"Value": "539454"}]]
     }
+
+
+def test_write_json_categoricals() -> None:
+    data = {"column": ["test1", "test2", "test3", "test4"]}
+    df = pl.DataFrame(data).with_columns(pl.col("column").cast(pl.Categorical))
+
+    assert (
+        df.write_json(row_oriented=True, file=None)
+        == '[{"column":"test1"},{"column":"test2"},{"column":"test3"},{"column":"test4"}]'
+    )