format code with black/fix unit test

unionai-oss · Jan 14, 2025 · 55bcbfa · 55bcbfa
1 parent a049763
commit 55bcbfa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/core/test_from_to_format_conversions.py b/tests/core/test_from_to_format_conversions.py
@@ -68,6 +68,7 @@ class InSchemaJsonNormalize(InSchema):
     class Config:
         from_format = "json_normalize"
 
+
 class OutSchema(InSchema):
     float_col: pa.typing.Series[float]
 
@@ -198,7 +199,7 @@ def _needs_pyarrow(schema) -> bool:
         [InSchemaParquet, lambda df, x: df.to_parquet(x), io.BytesIO],
         [InSchemaPickle, lambda df, x: df.to_pickle(x), io.BytesIO],
         [InSchemaPickleCallable, lambda df, x: df.to_pickle(x), io.BytesIO],
-        [InSchemaJsonNormalize, lambda df: df.to_json(), io.StringIO],
+        [InSchemaJsonNormalize, lambda df: df.to_dict(orient="records"), None],
     ],
 )
 def test_from_format(schema, to_fn, buf_cls):