fixed test_pickle_safe

madsbk · Apr 16, 2021 · 6548968 · 6548968
1 parent 084680c
commit 6548968
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/distributed/tests/test_publish.py b/distributed/tests/test_publish.py
@@ -1,4 +1,5 @@
 import asyncio
+from types import SimpleNamespace
 
 import pytest
 
@@ -271,12 +272,11 @@ async def test_pickle_safe(c, s, a, b):
         assert result == [1, 2, 3]
 
         with pytest.raises(TypeError):
-            await c2.publish_dataset(y=lambda x: x)
+            # SimpleNamespace() is not serializable
+            await c2.publish_dataset(y=SimpleNamespace())
 
         await c.publish_dataset(z=lambda x: x)  # this can use pickle
-
-        with pytest.raises(TypeError):
-            await c2.get_dataset("z")
+        await c2.get_dataset("z")
 
 
 @gen_cluster(client=True)