diff --git a/comfyui.py b/comfyui.py index 84b43dc..bbf3344 100644 --- a/comfyui.py +++ b/comfyui.py @@ -66,6 +66,7 @@ specific_test = os.getenv("SPECIFIC_TEST") if os.getenv("SPECIFIC_TEST") else "" skip_tests = os.getenv("SKIP_TESTS") +# Run a bunch of checks to verify input args: if test_all and specific_test: print(f"WARNING: can't have both TEST_ALL and SPECIFIC_TEST at the same time...") print(f"Running TEST_ALL instead") @@ -80,6 +81,11 @@ print(f"skip_tests: {skip_tests}") print("========================================") +if not test_workflows and workspace_name and not test_all: + print("\n!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!") + print("!!!! WARNING: You are deploying a workspace without TEST_ALL !!!!") + print("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!\n") + def install_comfyui(): snapshot = json.load(open("/root/workspace/snapshot.json", 'r')) comfyui_commit_sha = snapshot["comfyui"] @@ -361,6 +367,7 @@ def test_workflows(self): tests = [f"/root/workspace/workflows/{workflow}/{specific_test}"] else: tests = [f"/root/workspace/workflows/{workflow}/test.json"] + print("\n\n-----------------------------------------------------------") print(f"====> Running tests for {workflow}: ", tests) for test in tests: tool = Tool.from_yaml(f"/root/workspace/workflows/{workflow}/api.yaml")