Fix typing in test_stateful

Author: dstansby

pyproject.toml

@@ -355,6 +355,7 @@ module = [
     "tests.test_store.test_core",
     "tests.test_store.test_logging",
     "tests.test_store.test_object",
+    "tests.test_store.test_stateful",
 ]
 strict = false
 
@@ -363,7 +364,6 @@ strict = false
 [[tool.mypy.overrides]]
 module = [
     "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

@@ -16,18 +16,18 @@
 
 
 @pytest.mark.filterwarnings("ignore::zarr.core.dtype.common.UnstableSpecificationWarning")
-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):
@@ -38,4 +38,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]