Merge pull request #6 from neptyneco/widgets-unit-test

Fix encoding
neptyneco · Nov 12, 2024 · 97dd284 · 97dd284
2 parents 9e31640 + 68e5770
commit 97dd284
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/neptyne_kernel/json_tools.py b/neptyne_kernel/json_tools.py
@@ -16,14 +16,13 @@ def json_clean(obj: Any) -> dict:
 def json_default(obj: Any) -> str | int | list | float | None:
     if isinstance(obj, Empty):
         return None
+    if hasattr(obj, "__json__"):
+        return obj.__json__()
     try:
         from jupyter_client.jsonutil import json_default as jupyter_json_default
     except ImportError:
         from jupyter_client.jsonutil import date_default as jupyter_json_default
 
-    if hasattr(obj, "__json__"):
-        obj = obj.__json__()
-
     return jupyter_json_default(obj)
 
 

diff --git a/neptyne_kernel/widgets/register_widget_test.py b/neptyne_kernel/widgets/register_widget_test.py
@@ -4,6 +4,7 @@
 from typing import Callable
 
 from ..cell_range import CellRange
+from ..json_tools import json_default
 from ..neptyne_protocol import WidgetParamType
 from ..util import list_like
 from .base_widget import BaseWidget
@@ -115,7 +116,7 @@ def test_register_widget():
 def test_widgets_with_enums_can_serialize():
     scatter = Scatter(x=[1, 2, 3], y=[4, 5, 6], trendline=Scatter.TrendlineType.POLY)
 
-    s = json.dumps(scatter)
+    s = json.dumps(scatter, default=json_default)
     scatter2 = Scatter(**json.loads(s))
 
     assert scatter2.trendline == Scatter.TrendlineType.POLY