diff --git a/Makefile b/Makefile index 945991b2..c3180797 100644 --- 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 index c79b42d7..04dee326 100644 --- 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: