mob next [ci-skip] [ci skip] [skip ci]

lastFile:tests/unit/test_project.py

src/skore/project.py

@@ -44,7 +44,7 @@ def transform(o: Any) -> Item:
                 raw_class_name="sklearn.base.BaseEstimator",
                 serialized=json.dumps(
                     {
-                        "skops": skops.io.dumps(o),
+                        "skops": skops.io.dumps(o).decode(),
                         "html": sklearn.utils.estimator_html_repr(o),
                     }
                 ),

tests/unit/test_project.py

@@ -1,4 +1,8 @@
+import json
 import pandas
+import numpy
+import sklearn.svm
+import skops.io
 from skore.project import Item, transform
 
 
@@ -10,12 +14,41 @@ def test_transform_primitive():
 
 
 def test_transform_pandas_dataframe():
-    o = pandas.DataFrame()
+    o = pandas.DataFrame([{"key": "value"}])
     actual = transform(o)
     expected = Item(
-        raw=pandas.DataFrame(), raw_class_name="pandas.DataFrame", serialized=""
+        raw=o,
+        raw_class_name="pandas.DataFrame",
+        serialized=o.to_json(orient="split"),
     )
+
+    assert actual == expected
+
+
+def test_transform_numpy_ndarray():
+    o = numpy.ndarray([1, 2, 3])
+    actual = transform(o)
+    expected = Item(
+        raw=o,
+        raw_class_name="numpy.ndarray",
+        serialized=json.dumps(o.tolist()),
+    )
+
     assert actual == expected
 
-    # o = 3.3
-    # transform(o)
+
+def test_transform_sklearn_base_baseestimator():
+    o = sklearn.svm.SVC()
+    actual = transform(o)
+    expected = Item(
+        raw=o,
+        raw_class_name="numpy.ndarray",
+        serialized=json.dumps(
+            {
+                "skops": skops.io.dumps(o).decode(),
+                "html": sklearn.utils.estimator_html_repr(o),
+            }
+        ),
+    )
+
+    assert actual == expected