Add tests

Signed-off-by: Kevin Su <[email protected]>
flyteorg · May 5, 2023 · 85ccfad · 85ccfad
1 parent 0862652
commit 85ccfad
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/flytekit/unit/cli/pyflyte/test_run.py b/tests/flytekit/unit/cli/pyflyte/test_run.py
@@ -67,6 +67,16 @@ def test_imperative_wf():
     assert result.exit_code == 0
 
 
+def test_copy_all_files():
+    runner = CliRunner()
+    result = runner.invoke(
+        pyflyte.main,
+        ["run", "--copy-all", IMPERATIVE_WORKFLOW_FILE, "wf", "--in1", "hello", "--in2", "world"],
+        catch_exceptions=False,
+    )
+    assert result.exit_code == 0
+
+
 def test_pyflyte_run_cli():
     runner = CliRunner()
     parquet_file = os.path.join(DIR_NAME, "testdata/df.parquet")