Merge pull request #124 from wfcommons/testing

WfChef Testing++

Author: henricasanova

tests/wfchef/test_wfchef.py

@@ -16,6 +16,7 @@
 import sys
 
 from tests.test_helpers import _create_fresh_local_dir
+from wfcommons.wfchef.chef import get_recipe
 from wfcommons.wfchef.chef import create_recipe
 from wfcommons.wfchef.chef import install_recipe
 from wfcommons.wfchef.chef import uninstall_recipe
@@ -123,6 +124,14 @@ def test_recipe_management_functions(self) -> None:
             sys.stderr.write(f"✗ Failed to use installed recipe: {e}\n")
             raise
 
+        try:
+            recipe = get_recipe("somename_recipe")
+            generator = WorkflowGenerator(recipe.from_num_tasks(250))
+        except Exception as e:
+            sys.stderr.write(f"✗ Failed to use get installed recipe by name: {e}\n")
+            raise
+
+        sys.stderr.write("✓ Successfully used recipe via direct import and by using get_recipe(name)\n")
 
         # Uninstall the recipe
         sys.stderr.write("\n" + "=" * 60 + "\n")

wfcommons/wfchef/chef.py

@@ -23,8 +23,9 @@
 import traceback
 import sys
 
-from typing import Dict, Optional, Union
+from typing import Dict, Optional, Union, Type
 from stringcase import capitalcase
+from wfcommons.wfchef.wfchef_abstract_recipe import WfChefWorkflowRecipe
 
 from .duplicate import duplicate, NoMicrostructuresError
 from .find_microstructures import save_microstructures
@@ -166,7 +167,7 @@ def analyzer_summary(path_to_instances: pathlib.Path) -> Dict:
     return stats_dict
 
 
-def get_recipe(recipe: str) -> Optional[type]:
+def get_recipe(recipe: str) -> Type[WfChefWorkflowRecipe] | None:
     """
     Load a recipe by name from installed entry points.