test the tflite commands

cpmpercussion · Aug 25, 2024 · 7fadcb1 · 7fadcb1
1 parent 3111929
commit 7fadcb1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_commands.py b/tests/test_commands.py
@@ -12,3 +12,9 @@ def test_model_test_command():
     runner = CliRunner()
     result = runner.invoke(cli, ["test-mdrnn"])
     # assert result.exit_code == 0
+
+def test_tflite_converter_command(models_location, keras_file, weights_file):
+    runner = CliRunner()
+    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)])