Adjust tests

jupyter-widgets · Aug 6, 2017 · f830665 · f830665
1 parent 35918bb
commit f830665
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/ipywidgets/widgets/tests/test_set_state.py b/ipywidgets/widgets/tests/test_set_state.py
@@ -45,7 +45,7 @@ def serializer(instance, widget):
     return { 'data': memoryview(instance.data) if instance.data else None }
 
 def deserializer(json_data, widget):
-    return DataInstance( byte_type(json_data['data']) if json_data else None )
+    return DataInstance( json_data['data'].tobytes() if json_data else None )
 
 class DataWidget(SimpleWidget):
     d = Instance(DataInstance).tag(sync=True, to_json=serializer, from_json=deserializer)
@@ -54,7 +54,7 @@ class DataWidget(SimpleWidget):
 
 # A widget that has a buffer that might be changed on reception:
 def truncate_deserializer(json_data, widget):
-    return DataInstance( byte_type(json_data['data'][:20]) if json_data else None )
+    return DataInstance( json_data['data'][:20].tobytes() if json_data else None )
 
 class TruncateDataWidget(SimpleWidget):
     d = Instance(DataInstance).tag(sync=True, to_json=serializer, from_json=truncate_deserializer)

diff --git a/ipywidgets/widgets/tests/utils.py b/ipywidgets/widgets/tests/utils.py
@@ -38,6 +38,7 @@ def teardown_test_comm():
             delattr(Widget, attr)
         else:
             setattr(Widget, attr, value)
+    _widget_attrs.clear()
 
 def setup():
     setup_test_comm()