Merge pull request #136 from wfcommons/testing

WfChef coverage++

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.utils import draw
 from wfcommons.wfchef.chef import get_recipe
 from wfcommons.wfchef.chef import create_recipe
 from wfcommons.wfchef.chef import install_recipe
@@ -124,6 +125,18 @@ def test_recipe_management_functions(self) -> None:
             sys.stderr.write(f"✗ Failed to use installed recipe: {e}\n")
             raise
 
+        # Test the graph drawing utility
+        sys.stderr.write("\n" + "=" * 60 + "\n")
+        sys.stderr.write("Testing the graph drawing utility...\n")
+        sys.stderr.write("=" * 60 + "\n")
+        try:
+            generator = WorkflowGenerator(SomenameRecipe.from_num_tasks(50))
+            workflow = generator.build_workflow()
+            plt_figure, plt_axis = draw(workflow) # Coverage
+        except Exception as e:
+            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))
@@ -301,8 +314,6 @@ def test_recipe_management_wfchef_main(self, monkeypatch, capsys) -> None:
             assert e.code == 0
         capsys.readouterr() # Clear output
 
-
-
         # Calling main with 'ls' command
         with capsys.disabled():
             sys.stderr.write("\n" + "=" * 60 + "\n")