run __main__ in test

biodigitalmatter · Nov 4, 2024 · 553b8d8 · 553b8d8
1 parent 0d3d9cb
commit 553b8d8
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 8 deletions.
diff --git a/src/bdm_voxel_builder/__main__.py b/src/bdm_voxel_builder/__main__.py
@@ -54,14 +54,7 @@ def _load_config(config_file: pathlib.Path) -> Config:
 
     return config_module.config
 
-
-@click.command()
-@click.argument(
-    "config-file", type=click.Path(exists=True), default=DATA_DIR / "config.py"
-)
-def main(config_file):
-    config: Config = _load_config(config_file)
-
+def run_algo(config: Config):
     algo = config.algo
     visualizer = config.visualizer
     iterations = config.iterations
@@ -92,5 +85,13 @@ def main(config_file):
             visualizer.show()
 
 
+@click.command()
+@click.argument(
+    "config-file", type=click.Path(exists=True), default=DATA_DIR / "config.py"
+)
+def main(config_file):
+    config: Config = _load_config(config_file)
+    run_algo(config)
+
 if __name__ == "__main__":
     main()
diff --git a/tests/test_run_with_config.py b/tests/test_run_with_config.py
@@ -0,0 +1,12 @@
+from bdm_voxel_builder import DATA_DIR
+from bdm_voxel_builder.__main__ import _load_config, run_algo
+from bdm_voxel_builder.config_setup import Config
+
+
+def test_run_with_config():  
+    config: Config = _load_config(DATA_DIR / "config.py")
+    config.iterations = 10
+    config.visualizer = None
+
+    run_algo(config)
+