Add test & type annotation

Author: MetRonnie

cylc/flow/data_store_mgr.py

@@ -349,7 +349,7 @@ def runtime_from_config(rtconfig):
     )
 
 
-def runtime_from_partial(rtconfig, runtimeold=None):
+def runtime_from_partial(rtconfig, runtimeold: Optional[PbRuntime] = None):
     """Populate runtime object from partial/full config.
 
     Potentially slower than the non-partial one, due to tha the setattr calls,

tests/unit/network/test_schema.py

@@ -96,15 +96,12 @@ def test_sort_args(elements, sort_args, expected_result):
         assert elements == expected_result
 
 
-@pytest.mark.parametrize(
-    'field_name', RUNTIME_FIELD_TO_CFG_MAP.keys()
-)
-def test_runtime_field_to_cfg_map(field_name: str):
+def test_runtime_field_to_cfg_map():
     """Ensure the Runtime type's fields can be mapped back to the workflow
     config."""
-    cfg_name = RUNTIME_FIELD_TO_CFG_MAP[field_name]
-    assert field_name in Runtime.__dict__
-    assert WORKFLOW_SPEC.get('runtime', '__MANY__', cfg_name)
+    assert set(RUNTIME_FIELD_TO_CFG_MAP) == set(Runtime._meta.fields)
+    for cfg_name in RUNTIME_FIELD_TO_CFG_MAP.values():
+        assert WORKFLOW_SPEC.get('runtime', '__MANY__', cfg_name)
 
 
 @pytest.mark.parametrize('runtime_dict,expected', [