Fix unit test

AlibabaPAI · Oct 17, 2024 · 268640d · 268640d
1 parent f9c6d43
commit 268640d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/Makefile b/Makefile
@@ -29,13 +29,13 @@ lint: check_pylint_installed check_pytest_installed
 
 .PHONY: test
 test: check_pytest_installed
-	@pytest -x -v --ignore=third_party/ --ignore=tests/e2e_test --disable-warnings
+	@pytest -v --ignore=third_party/ --ignore=tests/e2e_test --disable-warnings
 	@python examlpes/offline_inference.py
 	@pytest -v tests/e2e_test/test_e2e.py
 
 .PHONY: unit_test
 unit_test: check_pytest_installed
-	@pytest -x -v --ignore=third_party/ --ignore=tests/e2e_test --disable-warnings
+	@pytest -v --ignore=third_party/ --ignore=tests/e2e_test --disable-warnings
 
 .PHONY: offline_test
 offline_test:

diff --git a/tests/unit_test/llumlet/test_local_migration_scheduler.py b/tests/unit_test/llumlet/test_local_migration_scheduler.py
@@ -13,7 +13,7 @@
 
 import math
 from llumnix.llumlet.local_migration_scheduler import LocalMigrationScheduler
-from llumnix.llumlet.request import LlumnixRequest, RequestInferenceType
+from llumnix.llumlet.request import LlumnixRequest, RequestInferenceType, RequestStatus
 
 class MockRequest(LlumnixRequest):
     def __init__(self, request_id, length, expected_steps) -> None:
@@ -32,11 +32,11 @@ def inference_type(self) -> RequestInferenceType:
 
     @property
     def request_len(self) -> int:
-        pass
+        return self.length
 
     @property
     def prompt_len(self) -> int:
-        pass
+        return self.length
 
     @property
     def output_len(self) -> int:
@@ -48,7 +48,7 @@ def arrival_time(self) -> float:
 
     @property
     def status(self) -> RequestStatus:
-        pass
+        return self._status
 
     @property
     def prefill_num_blocks(self) -> int: