Skip to content

Commit 28f3092

Browse files
thiago-aixplainThiago Castro Ferreira
andauthored
Do not run service mode tests for specific pipelines (#159)
Co-authored-by: Thiago Castro Ferreira <[email protected]>
1 parent 8df2bd0 commit 28f3092

File tree

1 file changed

+25
-29
lines changed

1 file changed

+25
-29
lines changed

tests/functional/pipelines/run_test.py

Lines changed: 25 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -109,59 +109,55 @@ def test_run_multipipe_with_datasets(batchmode: bool):
109109
assert response["status"] == "SUCCESS"
110110

111111

112-
@pytest.mark.parametrize("batchmode", [True, False])
113-
def test_run_segment_reconstruct(batchmode: bool):
112+
def test_run_segment_reconstruct():
114113
pipeline = PipelineFactory.list(query="Segmentation/Reconstruction Functional Test - DO NOT DELETE")["results"][0]
115-
response = pipeline.run("https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav", **{"batchmode": batchmode})
114+
response = pipeline.run("https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav")
116115

117116
assert response["status"] == "SUCCESS"
118117
output = response["data"][0]
119118
assert output["label"] == "Output 1"
120119

121120

122-
@pytest.mark.parametrize("batchmode", [True, False])
123-
def test_run_metric(batchmode: bool):
121+
def test_run_metric():
124122
pipeline = PipelineFactory.list(query="ASR Metric Functional Test - DO NOT DELETE")["results"][0]
125-
response = pipeline.run({
126-
"AudioInput": "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav",
127-
"ReferenceInput": "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.txt"
128-
}, **{"batchmode": batchmode})
129-
123+
response = pipeline.run(
124+
{
125+
"AudioInput": "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav",
126+
"ReferenceInput": "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.txt",
127+
}
128+
)
129+
130130
assert response["status"] == "SUCCESS"
131131
assert len(response["data"]) == 2
132132
assert response["data"][0]["label"] in ["TranscriptOutput", "ScoreOutput"]
133133
assert response["data"][1]["label"] in ["TranscriptOutput", "ScoreOutput"]
134134

135135

136136
@pytest.mark.parametrize(
137-
"batchmode,input_data,output_data",
137+
"input_data,output_data",
138138
[
139-
(True, "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav", "AudioOutput"),
140-
(False, "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav", "AudioOutput"),
141-
(True, "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.txt", "TextOutput"),
142-
(False, "https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.txt", "TextOutput")
143-
]
139+
("https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.wav", "AudioOutput"),
140+
("https://aixplain-platform-assets.s3.amazonaws.com/samples/en/CPAC1x2.txt", "TextOutput"),
141+
],
144142
)
145-
def test_run_router(batchmode: bool, input_data: str, output_data: str):
143+
def test_run_router(input_data: str, output_data: str):
146144
pipeline = PipelineFactory.list(query="Router Test - DO NOT DELETE")["results"][0]
147-
response = pipeline.run(input_data, **{"batchmode": batchmode})
148-
145+
response = pipeline.run(input_data)
146+
149147
assert response["status"] == "SUCCESS"
150148
assert response["data"][0]["label"] == output_data
151149

152150

153151
@pytest.mark.parametrize(
154-
"batchmode,input_data,output_data",
152+
"input_data,output_data",
155153
[
156-
(True, "I love it.", "PositiveOutput"),
157-
(False, "I love it.", "PositiveOutput"),
158-
(True, "I hate it.", "NegativeOutput"),
159-
(False, "I hate it.", "NegativeOutput")
160-
]
154+
("I love it.", "PositiveOutput"),
155+
("I hate it.", "NegativeOutput"),
156+
],
161157
)
162-
def test_run_decision(batchmode: bool, input_data: str, output_data: str):
158+
def test_run_decision(input_data: str, output_data: str):
163159
pipeline = PipelineFactory.list(query="Decision Test - DO NOT DELETE")["results"][0]
164-
response = pipeline.run(input_data, **{"batchmode": batchmode})
165-
160+
response = pipeline.run(input_data)
161+
166162
assert response["status"] == "SUCCESS"
167-
assert response["data"][0]["label"] == output_data
163+
assert response["data"][0]["label"] == output_data

0 commit comments

Comments
 (0)