lkva/NDOJ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix pretest wrong testcases

Browse source
This commit is contained in:
cuom1999 2020-05-29 15:46:17 -05:00
parent 6da7630ba5
commit 388715bf3a
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
judge/views/submission.py
View file

@ -165,11 +165,17 @@ def get_problem_data(submission):
archive = zipfile.ZipFile(archive_path, 'r') archive = zipfile.ZipFile(archive_path, 'r')
except zipfile.BadZipfile: except zipfile.BadZipfile:
raise Exception('bad archive: "%s"' % archive_path) raise Exception('bad archive: "%s"' % archive_path)
testcases = ProblemTestCase.objects.filter(dataset=submission.problem)\
.order_by('order')
problem_data = {case.order: get_input_answer(case, archive) testcases = ProblemTestCase.objects.filter(dataset=submission.problem)\
for case in testcases} .order_by('order')
if (submission.is_pretested):
testcases = testcases.filter(is_pretest=True)
problem_data = {}
for count, case in enumerate(testcases):
problem_data[count + 1] = get_input_answer(case, archive)
return problem_data return problem_data