Skip to content

Commit 6e7ea76

Browse files
committed
tests actually need to be prefixed with 'test'
Signed-off-by: Sophie du Couédic <[email protected]>
1 parent c4c8ca8 commit 6e7ea76

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

tests/e2e/test_spyre_cb_inference_steps.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,8 @@
1414
@pytest.mark.cb
1515
@pytest.mark.parametrize("model", get_spyre_model_list())
1616
@pytest.mark.parametrize("backend", get_spyre_backend_list())
17-
def prompts_aligned_with_tkv_boundaries(model: str, backend: str,
18-
monkeypatch: pytest.MonkeyPatch):
17+
def test_prompts_aligned_with_tkv_boundaries(model: str, backend: str,
18+
monkeypatch: pytest.MonkeyPatch):
1919
""" Scenario where it happens that all the sequences get scheduled in a
2020
fashion where they are aligned with the block boundaries (i.e. tkv multiple
2121
of 64 at the time of prefilling).
@@ -176,8 +176,8 @@ def prompts_aligned_with_tkv_boundaries(model: str, backend: str,
176176
@pytest.mark.cb
177177
@pytest.mark.parametrize("model", get_spyre_model_list())
178178
@pytest.mark.parametrize("backend", get_spyre_backend_list())
179-
def prompts_misaligned_with_tkv_boundaries(model: str, backend: str,
180-
monkeypatch: pytest.MonkeyPatch):
179+
def test_prompts_misaligned_with_tkv_boundaries(
180+
model: str, backend: str, monkeypatch: pytest.MonkeyPatch):
181181
""" Scenario where it happens that some sequence gets scheduled in a way
182182
that it is misaligned with the block boundary (i.e. tkv is not a multiple
183183
of 64 at the time of prefilling).
@@ -338,7 +338,7 @@ def prompts_misaligned_with_tkv_boundaries(model: str, backend: str,
338338
@pytest.mark.cb
339339
@pytest.mark.parametrize("model", get_spyre_model_list())
340340
@pytest.mark.parametrize("backend", get_spyre_backend_list())
341-
def two_sequences_finish_same_time_as_new_arrive(
341+
def test_two_sequences_finish_same_time_as_new_arrive(
342342
model: str, backend: str, monkeypatch: pytest.MonkeyPatch):
343343
""" 2-cases-in-1: (1) Two sequences finish at the same time and (2) a new
344344
request arrives when another finishes.
@@ -476,8 +476,8 @@ def two_sequences_finish_same_time_as_new_arrive(
476476
@pytest.mark.cb
477477
@pytest.mark.parametrize("model", get_spyre_model_list())
478478
@pytest.mark.parametrize("backend", get_spyre_backend_list())
479-
def prompt_too_long_for_current_tkv(model: str, backend: str,
480-
monkeypatch: pytest.MonkeyPatch):
479+
def test_prompt_too_long_for_current_tkv(model: str, backend: str,
480+
monkeypatch: pytest.MonkeyPatch):
481481
""" Scenario where the requested prompt is too long for current tkv value
482482
483483
Configuration:
@@ -623,7 +623,7 @@ def prompt_too_long_for_current_tkv(model: str, backend: str,
623623
@pytest.mark.cb
624624
@pytest.mark.parametrize("model", get_spyre_model_list())
625625
@pytest.mark.parametrize("backend", get_spyre_backend_list())
626-
def requested_tokens_not_fitting_remaining_space(
626+
def test_requested_tokens_not_fitting_remaining_space(
627627
model: str, backend: str, monkeypatch: pytest.MonkeyPatch):
628628
""" Scenario where the request goes beyond max_model_len
629629
@@ -808,8 +808,8 @@ def requested_tokens_not_fitting_remaining_space(
808808
@pytest.mark.cb
809809
@pytest.mark.parametrize("model", get_spyre_model_list())
810810
@pytest.mark.parametrize("backend", get_spyre_backend_list())
811-
def requests_use_all_available_blocks(model: str, backend: str,
812-
monkeypatch: pytest.MonkeyPatch):
811+
def test_requests_use_all_available_blocks(model: str, backend: str,
812+
monkeypatch: pytest.MonkeyPatch):
813813
""" Scenario where the requests use all of the available blocks
814814
815815
Configuration:
@@ -939,8 +939,8 @@ def requests_use_all_available_blocks(model: str, backend: str,
939939
@pytest.mark.cb
940940
@pytest.mark.parametrize("model", get_spyre_model_list())
941941
@pytest.mark.parametrize("backend", get_spyre_backend_list())
942-
def requests_use_more_than_available_blocks(model: str, backend: str,
943-
monkeypatch: pytest.MonkeyPatch):
942+
def test_requests_use_more_than_available_blocks(
943+
model: str, backend: str, monkeypatch: pytest.MonkeyPatch):
944944
""" Scenario where some request need to wait because of the number of
945945
available blocks.
946946

0 commit comments

Comments
 (0)