@@ -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