Fix router test (#4483)

This commit is contained in:
Byron Hsu
2025-03-16 22:49:47 -07:00
committed by GitHub
parent 452db50808
commit 8cc300f536
4 changed files with 21 additions and 20 deletions

View File

@@ -10,6 +10,7 @@ import threading
import time
import unittest
from concurrent.futures import ThreadPoolExecutor
from dataclasses import dataclass
from functools import partial
from types import SimpleNamespace
from typing import Callable, List, Optional, Tuple
@@ -453,7 +454,13 @@ def run_with_timeout(
return ret_value[0]
def run_unittest_files(files: List, timeout_per_file: float):
@dataclass
class TestFile:
name: str
estimated_time: float = 60
def run_unittest_files(files: List[TestFile], timeout_per_file: float):
tic = time.time()
success = True