Skip to content

Commit

Permalink
add stored architecture test
Browse files Browse the repository at this point in the history
  • Loading branch information
mzouink committed Nov 13, 2024
1 parent 0d3b3d2 commit c35b702
Showing 1 changed file with 22 additions and 2 deletions.
24 changes: 22 additions & 2 deletions tests/operations/test_architecture.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}"

0 comments on commit c35b702

Please sign in to comment.