From 007f4f7314eabd9cc3a2b0d11889de49ad3c682a Mon Sep 17 00:00:00 2001
From: Vladimir Repin <32306715+mezotaken@users.noreply.github.com>
Date: Sat, 12 Nov 2022 15:12:15 +0300
Subject: Tests cleaned up

---
 launch.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'launch.py')

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()
-- 
cgit v1.2.3