Update tests to expect specific events

Author: krassowski

tests/test_ynotebook.py

@@ -2,7 +2,7 @@
 # Distributed under the terms of the Modified BSD License.
 
 
-from pycrdt import Map, MapEvent
+from pycrdt import ArrayEvent, Map, MapEvent, TextEvent
 from pytest import mark
 
 from jupyter_ydoc import YNotebook
@@ -119,17 +119,17 @@ def record_changes(topic, event):
 
 
 @mark.parametrize(
-    "modifications",
+    "modifications, expected_events",
     [
         # modifications of single attributes
-        [["source", "'b'"]],
-        [["outputs", []]],
-        [["execution_count", 2]],
+        ([["source", "'b'"]], {TextEvent}),
+        ([["outputs", []]], {ArrayEvent}),
+        ([["execution_count", 2]], {MapEvent}),
         # multi-attribute modifications
-        [["source", "10"], ["execution_count", 10]],
+        ([["source", "10"], ["execution_count", 10]], {TextEvent, MapEvent}),
     ],
 )
-def test_modify_single_cell(modifications):
+def test_modify_single_cell(modifications, expected_events):
     nb = YNotebook()
     nb.set(
         {
@@ -164,7 +164,9 @@ def record_changes(topic, event):
 
     for modification in modifications:
         key, new_value = modification
-        assert nb.ycells[0][key] == new_value
+        after = nb.ycells[0][key]
+        after_py = after.to_py() if hasattr(after, "to_py") else after
+        assert after_py == new_value
 
     # there should be only one change
     assert len(changes) == 1
@@ -173,6 +175,5 @@ def record_changes(topic, event):
     assert len(cell_events) == 1
     # but it should be a change to cell data, not a change to the cell list
     events = cell_events[0]
-    # (it's neseted for some reason)
-    assert len(events) == 1
-    assert isinstance(events[0], MapEvent)
+    assert len(events) == len(expected_events)
+    assert {type(e) for e in events} == expected_events