Add serialization fix

Author: eliotwrobson

src/pytest_pl_grader/_student_code_runner.py

@@ -72,7 +72,7 @@ def student_function_temp() -> Any:
 
     function_response: StudentFunctionResponse = {
         "status": "success" if execution_error is None else "exception",
-        "value": result,
+        "value": to_json(result),
         "stdout": stdout_capture.getvalue(),
         "stderr": stderr_capture.getvalue(),
         "exception_name": type(execution_error).__name__,

src/pytest_pl_grader/json_utils.py

@@ -13,6 +13,11 @@ def to_json(v: Any) -> Any:
             "_concrete_type": type(v).__name__,
             "_value": str(v),
         }
+    elif isinstance(v, np.bool_):
+        return {
+            "_type": "np_bool",
+            "_value": str(v),
+        }
     elif isinstance(v, np.ndarray):
         if np.isrealobj(v):
             return {"_type": "ndarray", "_value": v.tolist(), "_dtype": str(v.dtype)}
@@ -56,6 +61,11 @@ def from_json(v_json) -> Any:
                 return complex(v_json["_value"]["real"], v_json["_value"]["imag"])
             else:
                 raise ValueError("variable of type complex should have value with real and imaginary pair")
+        elif v_json["_type"] == "np_bool":
+            if "_value" in v_json:
+                return np.bool_(v_json["_value"] == "True")
+            else:
+                raise ValueError("variable of type np_bool should have value")
         elif v_json["_type"] == "np_scalar":
             if "_concrete_type" in v_json and "_value" in v_json:
                 return getattr(np, v_json["_concrete_type"])(v_json["_value"])

tests/test_serialize.py

@@ -24,7 +24,7 @@ def test_serialize_numpy_array() -> None:
     assert np.array_equal(arr, deserialized)
 
 
-@pytest.mark.parametrize("obj", [np.bool(True), np.int32(42), np.float64(3.14), complex(1, 2)])
+@pytest.mark.parametrize("obj", [np.bool(True), np.int32(42), np.float64(3.14), complex(1, 2), True])
 def test_serialize_json(obj: Any) -> None:
     # Serialize the object to JSON-compatible format
     json_compatible = to_json(obj)