blocked out some basic tests

cpmpercussion · Aug 25, 2024 · f0e1120 · f0e1120
1 parent 7fadcb1
commit f0e1120
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/test_commands.py b/tests/test_commands.py
@@ -18,3 +18,19 @@ def test_tflite_converter_command(models_location, keras_file, weights_file):
     result = runner.invoke(cli, ["convert-tflite", "--out_dir", str(models_location)])
     result = runner.invoke(cli, ["convert-tflite", "-model", str(keras_file),"--out_dir", str(models_location)])
     result = runner.invoke(cli, ["convert-tflite", "-model", str(weights_file),"--out_dir", str(models_location)])
+
+# def test_train_command():
+    # runner = CliRunner()
+    # result = runner.invoke(cli, ["train", "--out_dir", str(models_location)])
+
+# def test_dataset_command():
+#     runner = CliRunner()
+#     result = runner.invoke(cli, ["dataset"])
+
+def test_run_command():
+    runner = CliRunner()
+    result = runner.invoke(cli, ["run"])
+
+def test_webui_command():
+    runner = CliRunner()
+    result = runner.invoke(cli, ["webui"])