Tests cleaned up

1 files changed, 4 insertions, 1 deletions

diff --git a/launch.py b/launch.py
index 8e65676d..6822a01d 100644
--- a/launch.py
+++ b/launch.py
@@ -229,6 +229,9 @@ def prepare_enviroment():
 def tests(argv):

     if "--api" not in argv:

         argv.append("--api")

+    if "--ckpt" not in argv:

+        argv.append("--ckpt")

+        argv.append("./test/test_files/empty.pt")

 

     print(f"Launching Web UI in another process for testing with arguments: {' '.join(argv[1:])}")

 

@@ -236,7 +239,7 @@ def tests(argv):
         proc = subprocess.Popen([sys.executable, *argv], stdout=stdout, stderr=stderr)

 

     import test.server_poll

-    test.server_poll.run_tests()

+    test.server_poll.run_tests(proc)

 

     print(f"Stopping Web UI process with id {proc.pid}")

     proc.kill()