Fix test_stateful

Author: dstansby

pyproject.toml

@@ -367,6 +367,7 @@ module = [
     "tests.test_store.test_core",
     "tests.test_store.test_logging",
     "tests.test_store.test_object",
+    "tests.test_store.test_stateful",
 ]
 strict = false
 
@@ -376,7 +377,6 @@ strict = false
 module = [
     "tests.test_codecs.test_codecs",
     "tests.test_metadata.*",
-    "tests.test_store.test_stateful",
     "tests.test_store.test_wrapper",
     "tests.test_group",
     "tests.test_indexing",

tests/test_store/test_stateful.py

@@ -15,18 +15,18 @@
 ]
 
 
-def test_zarr_hierarchy(sync_store: Store):
+def test_zarr_hierarchy(sync_store: Store) -> None:
     def mk_test_instance_sync() -> ZarrHierarchyStateMachine:
         return ZarrHierarchyStateMachine(sync_store)
 
     if isinstance(sync_store, ZipStore):
         pytest.skip(reason="ZipStore does not support delete")
 
-    run_state_machine_as_test(mk_test_instance_sync)
+    run_state_machine_as_test(mk_test_instance_sync)  # type ignore[no-untyped-call]
 
 
 def test_zarr_store(sync_store: Store) -> None:
-    def mk_test_instance_sync() -> None:
+    def mk_test_instance_sync() -> ZarrStoreStateMachine:
         return ZarrStoreStateMachine(sync_store)
 
     if isinstance(sync_store, ZipStore):
@@ -37,4 +37,4 @@ def mk_test_instance_sync() -> None:
         # It assumes that `set` and `delete` are the only two operations that modify state.
         # But LocalStore, directories can hang around even after a key is delete-d.
         pytest.skip(reason="Test isn't suitable for LocalStore.")
-    run_state_machine_as_test(mk_test_instance_sync)
+    run_state_machine_as_test(mk_test_instance_sync)  # type ignore[no-untyped-call]