22
33import judge0
44import pytest
5- from judge0 import Status , Submission , TestCase
65from judge0 .api import create_submissions_from_test_cases
6+ from judge0 .base_types import LanguageAlias , Status , TestCase
7+ from judge0 .submission import Submission
78
89
910@pytest .mark .parametrize (
@@ -172,14 +173,20 @@ def test_single_test_case_in_iterable(self, test_cases, stdin, expected_output):
172173 [Status .ACCEPTED , Status .ACCEPTED ],
173174 ],
174175 [
175- Submission (source_code = "print(f'Hello, {input()}')" ),
176+ Submission (
177+ source_code = "print(f'Hello, {input()}')" ,
178+ language = LanguageAlias .PYTHON ,
179+ ),
176180 [
177181 TestCase ("Judge0" , "Hello, Judge0" ),
178182 ],
179183 [Status .ACCEPTED ],
180184 ],
181185 [
182- Submission (source_code = "print(f'Hello, {input()}')" ),
186+ Submission (
187+ source_code = "print(f'Hello, {input()}')" ,
188+ language = LanguageAlias .PYTHON ,
189+ ),
183190 [
184191 TestCase ("Judge0" , "Hello, Judge0" ),
185192 TestCase ("pytest" , "Hi, pytest" ),
@@ -188,8 +195,14 @@ def test_single_test_case_in_iterable(self, test_cases, stdin, expected_output):
188195 ],
189196 [
190197 [
191- Submission (source_code = "print(f'Hello, {input()}')" ),
192- Submission (source_code = "print(f'Hello, {input()}')" ),
198+ Submission (
199+ source_code = "print(f'Hello, {input()}')" ,
200+ language = LanguageAlias .PYTHON ,
201+ ),
202+ Submission (
203+ source_code = "print(f'Hello, {input()}')" ,
204+ language = LanguageAlias .PYTHON ,
205+ ),
193206 ],
194207 [
195208 TestCase ("Judge0" , "Hello, Judge0" ),
@@ -207,13 +220,14 @@ def test_single_test_case_in_iterable(self, test_cases, stdin, expected_output):
207220def test_test_cases_from_run (
208221 source_code_or_submissions , test_cases , expected_status , request
209222):
210- client = request .getfixturevalue ("default_ce_client " )
223+ client = request .getfixturevalue ("ce_client " )
211224
212225 if isinstance (source_code_or_submissions , str ):
213226 submissions = judge0 .run (
214227 client = client ,
215228 source_code = source_code_or_submissions ,
216229 test_cases = test_cases ,
230+ language = LanguageAlias .PYTHON ,
217231 )
218232 else :
219233 submissions = judge0 .run (
@@ -231,6 +245,7 @@ def test_test_cases_from_run(
231245 [
232246 Submission (
233247 source_code = "print(f'Hello, {input()}')" ,
248+ language = LanguageAlias .PYTHON ,
234249 stdin = "Judge0" ,
235250 expected_output = "Hello, Judge0" ,
236251 ),
@@ -240,11 +255,13 @@ def test_test_cases_from_run(
240255 [
241256 Submission (
242257 source_code = "print(f'Hello, {input()}')" ,
258+ language = LanguageAlias .PYTHON ,
243259 stdin = "Judge0" ,
244260 expected_output = "Hello, Judge0" ,
245261 ),
246262 Submission (
247263 source_code = "print(f'Hello, {input()}')" ,
264+ language = LanguageAlias .PYTHON ,
248265 stdin = "pytest" ,
249266 expected_output = "Hello, pytest" ,
250267 ),
@@ -254,7 +271,7 @@ def test_test_cases_from_run(
254271 ],
255272)
256273def test_no_test_cases (submissions , expected_status , request ):
257- client = request .getfixturevalue ("default_ce_client " )
274+ client = request .getfixturevalue ("ce_client " )
258275
259276 submissions = judge0 .run (
260277 client = client ,
@@ -269,9 +286,13 @@ def test_no_test_cases(submissions, expected_status, request):
269286
270287@pytest .mark .parametrize ("n_submissions" , [42 , 84 ])
271288def test_batched_test_cases (n_submissions , request ):
272- client = request .getfixturevalue ("default_ce_client " )
289+ client = request .getfixturevalue ("ce_client " )
273290 submissions = [
274- Submission (source_code = f"print({ i } )" , expected_output = f"{ i } " )
291+ Submission (
292+ source_code = f"print({ i } )" ,
293+ language = LanguageAlias .PYTHON ,
294+ expected_output = f"{ i } " ,
295+ )
275296 for i in range (n_submissions )
276297 ]
277298
0 commit comments