Fix test cases

Author: Hmnt39

tests/test_unstable/test_throttle.py

@@ -36,7 +36,6 @@ def task(task_id: int) -> int:
                 concurrent_count += 1
                 max_observed = max(max_observed, concurrent_count)
 
-            # Simulate work
             time.sleep(0.1)
 
             with lock:
@@ -50,7 +49,6 @@ def task(task_id: int) -> int:
 
     assert len(results) == 10
     assert max_observed <= max_concurrent
-    assert max_observed == max_concurrent
 
 
 def test_throttle_serial_execution() -> None:
@@ -61,20 +59,19 @@ def test_throttle_serial_execution() -> None:
     def serial_task(task_id: int) -> None:
         with throttle_serial.lease():
             with lock:
-                execution_order.append(f"start-{task_id}")
+                execution_order.append(task_id)
             time.sleep(0.05)
             with lock:
-                execution_order.append(f"end-{task_id}")
+                execution_order.append(task_id)
 
     with ThreadPoolExecutor(max_workers=3) as executor:
         futures = [executor.submit(serial_task, i) for i in range(3)]
         for f in as_completed(futures):
             f.result()
 
     for i in range(0, len(execution_order) - 1, 2):
-        task_id = execution_order[i].split("-")[1]
-        assert execution_order[i] == f"start-{task_id}"
-        assert execution_order[i + 1] == f"end-{task_id}"
+        task_id = execution_order[i]
+        assert execution_order[i + 1] == task_id
 
 
 def test_throttle_high_concurrency() -> None: