pickle support is not implemented for memory store

jhamman · Sep 17, 2024 · 6d7baf1 · 6d7baf1
1 parent 6b6cc3a
commit 6d7baf1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/v3/test_store/test_memory.py b/tests/v3/test_store/test_memory.py
@@ -92,3 +92,13 @@ def test_store_supports_partial_writes(self, store: GpuMemoryStore) -> None:
 
     def test_list_prefix(self, store: GpuMemoryStore) -> None:
         assert True
+
+    def test_serizalizable_store(self, store: MemoryStore) -> None:
+        with pytest.raises(NotImplementedError):
+            store.__getstate__()
+
+        with pytest.raises(NotImplementedError):
+            store.__setstate__({})
+
+        with pytest.raises(NotImplementedError):
+            pickle.dumps(store)