michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

test/server_poll.py: use absolute path for test

Browse source 
test/server_poll.py: fix absolute path
This commit is contained in:
Zhang Hua 2023-03-11 13:27:10 +08:00
parent 8e0d16e746
commit 9abe2f5e74
No known key found for this signature in database
GPG key ID: 69528D6DCD2BE030
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
test/server_poll.py
View file

@ -1,6 +1,8 @@
import unittest import unittest
import requests import requests
import time import time
import os
from modules.paths import script_path
def run_tests(proc, test_dir): def run_tests(proc, test_dir):
@ -15,8 +17,8 @@ def run_tests(proc, test_dir):
break break
if proc.poll() is None: if proc.poll() is None:
if test_dir is None: if test_dir is None:
test_dir = "test" test_dir = os.path.join(script_path, "test")
suite = unittest.TestLoader().discover(test_dir, pattern="*_test.py", top_level_dir="test") suite = unittest.TestLoader().discover(test_dir, pattern="*_test.py", top_level_dir=test_dir)
result = unittest.TextTestRunner(verbosity=2).run(suite) result = unittest.TextTestRunner(verbosity=2).run(suite)
return len(result.failures) + len(result.errors) return len(result.failures) + len(result.errors)
else: else: