diff --git a/tests/operations/test_architecture.py b/tests/operations/test_architecture.py index 5ba387b4..f1f20071 100644 --- a/tests/operations/test_architecture.py +++ b/tests/operations/test_architecture.py @@ -18,9 +18,29 @@ def test_architecture( architecture_config, ): + architecture = architecture_config.architecture_type(architecture_config) + assert architecture.dims is not None, f"Architecture dims are None {architecture}" + + + + +@pytest.mark.parametrize( + "architecture_config", + [ + lf("dummy_architecture"), + lf("unet_architecture"), + ], +) +def test_stored_architecture( + architecture_config, +): + from dacapo.store.create_store import create_config_store + config_store = create_config_store() + config_store.store_architecture_config(architecture_config) + + retrieved_arch_config = config_store.retrieve_architecture_config(architecture_config.name) - architecture_type = architecture_config.architecture_type - architecture = architecture_type(architecture_config) + architecture = retrieved_arch_config.architecture_type(retrieved_arch_config) assert architecture.dims is not None, f"Architecture dims are None {architecture}"