Spaces:
Runtime error
Runtime error
| import unittest | |
| import requests | |
| import time | |
| def run_tests(proc, test_dir): | |
| timeout_threshold = 240 | |
| start_time = time.time() | |
| while time.time()-start_time < timeout_threshold: | |
| try: | |
| requests.head("http://localhost:7860/") | |
| break | |
| except requests.exceptions.ConnectionError: | |
| if proc.poll() is not None: | |
| break | |
| if proc.poll() is None: | |
| if test_dir is None: | |
| test_dir = "test" | |
| suite = unittest.TestLoader().discover(test_dir, pattern="*_test.py", top_level_dir="test") | |
| result = unittest.TextTestRunner(verbosity=2).run(suite) | |
| return len(result.failures) + len(result.errors) | |
| else: | |
| print("Launch unsuccessful") | |
| return 1 | |