Mercurial > ~astiob > upreckon > hgweb
annotate problem.py @ 79:ee8a99dcaaed
Renamed configuration variable tasknames to problems
| author | Oleg Oshmyan <chortos@inbox.lv> |
|---|---|
| date | Mon, 17 Jan 2011 11:11:01 +0000 |
| parents | 69eadc60f4e2 |
| children | 06356af50bf9 |
| rev | line source |
|---|---|
| 21 | 1 #! /usr/bin/env python |
|
77
69eadc60f4e2
Memory limit is now applied to the RSS when os.wait4 is available
Oleg Oshmyan <chortos@inbox.lv>
parents:
76
diff
changeset
|
2 # Copyright (c) 2010-2011 Chortos-2 <chortos@inbox.lv> |
| 16 | 3 |
| 21 | 4 from __future__ import division, with_statement |
| 5 | |
| 6 try: | |
| 7 from compat import * | |
| 8 import config, testcases | |
| 9 except ImportError: | |
| 10 import __main__ | |
| 11 __main__.import_error(sys.exc_info()[1]) | |
| 12 else: | |
| 22 | 13 from __main__ import clock, options |
| 21 | 14 |
| 22 | 15 import os, re, sys |
| 21 | 16 |
| 16 | 17 try: |
|
75
007f7eb6fb2b
The test context stack is now a deque.
Oleg Oshmyan <chortos@inbox.lv>
parents:
74
diff
changeset
|
18 from collections import deque |
|
007f7eb6fb2b
The test context stack is now a deque.
Oleg Oshmyan <chortos@inbox.lv>
parents:
74
diff
changeset
|
19 except ImportError: |
|
007f7eb6fb2b
The test context stack is now a deque.
Oleg Oshmyan <chortos@inbox.lv>
parents:
74
diff
changeset
|
20 deque = list |
|
007f7eb6fb2b
The test context stack is now a deque.
Oleg Oshmyan <chortos@inbox.lv>
parents:
74
diff
changeset
|
21 |
|
007f7eb6fb2b
The test context stack is now a deque.
Oleg Oshmyan <chortos@inbox.lv>
parents:
74
diff
changeset
|
22 try: |
| 21 | 23 import signal |
| 24 except ImportError: | |
| 25 signalnames = () | |
| 26 else: | |
| 27 # Construct a cache of all signal names available on the current | |
| 28 # platform. Prefer names from the UNIX standards over other versions. | |
| 29 unixnames = frozenset(('HUP', 'INT', 'QUIT', 'ILL', 'ABRT', 'FPE', 'KILL', 'SEGV', 'PIPE', 'ALRM', 'TERM', 'USR1', 'USR2', 'CHLD', 'CONT', 'STOP', 'TSTP', 'TTIN', 'TTOU', 'BUS', 'POLL', 'PROF', 'SYS', 'TRAP', 'URG', 'VTALRM', 'XCPU', 'XFSZ')) | |
| 30 signalnames = {} | |
| 31 for name in dir(signal): | |
| 32 if re.match('SIG[A-Z]+$', name): | |
| 33 value = signal.__dict__[name] | |
| 22 | 34 if isinstance(value, int) and (value not in signalnames or name[3:] in unixnames): |
| 21 | 35 signalnames[value] = name |
| 36 del unixnames | |
| 16 | 37 |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
38 __all__ = 'Problem', 'TestContext', 'test_context_end', 'TestGroup' |
| 21 | 39 |
| 26 | 40 def strerror(e): |
| 41 s = getattr(e, 'strerror') | |
| 42 if not s: s = str(e) | |
| 43 return ' (%s%s)' % (s[0].lower(), s[1:]) if s else '' | |
| 21 | 44 |
| 45 class Cache(object): | |
| 46 def __init__(self, mydict): | |
| 47 self.__dict__ = mydict | |
| 16 | 48 |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
49 class TestContext(object): |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
50 pass |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
51 |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
52 test_context_end = object() |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
53 |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
54 class TestGroup(TestContext): |
|
76
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
55 __slots__ = 'points', 'case', 'log', 'correct', 'allcorrect', 'real', 'max', 'ntotal', 'nvalued', 'ncorrect', 'ncorrectvalued' |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
56 |
|
76
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
57 def __init__(self, points=None): |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
58 self.points = points |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
59 self.real = self.max = self.ntotal = self.nvalued = self.ncorrect = self.ncorrectvalued = 0 |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
60 self.allcorrect = True |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
61 self.log = [] |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
62 |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
63 def case_start(self, case): |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
64 self.case = case |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
65 self.correct = False |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
66 self.ntotal += 1 |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
67 if case.points: |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
68 self.nvalued += 1 |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
69 |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
70 def case_correct(self): |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
71 self.correct = True |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
72 self.ncorrect += 1 |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
73 if self.case.points: |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
74 self.ncorrectvalued += 1 |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
75 |
|
76
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
76 def case_end(self): |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
77 self.log.append((self.case, self.correct)) |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
78 del self.case |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
79 if not self.correct: |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
80 self.allcorrect = False |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
81 |
|
76
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
82 def score(self, real, max): |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
83 self.real += real |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
84 self.max += max |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
85 |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
86 def end(self): |
|
76
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
87 if not self.allcorrect: |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
88 self.real = 0 |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
89 if self.points is not None and self.points != self.max: |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
90 max, weighted = self.points, self.real * self.points / self.max if self.max else 0 |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
91 before_weighting = ' (%g/%g before weighting)' % (self.real, self.max) |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
92 else: |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
93 max, weighted = self.max, self.real |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
94 before_weighting = '' |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
95 say('Group total: %d/%d tests, %g/%g points%s' % (self.ncorrect, self.ntotal, weighted, max, before_weighting)) |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
96 # No real need to flush stdout, as it will anyway be flushed in a moment, |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
97 # when either the problem total or the next test case's ID is printed |
|
76
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
98 return weighted, max, self.log |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
99 |
| 16 | 100 class Problem(object): |
| 101 __slots__ = 'name', 'config', 'cache', 'testcases' | |
| 102 | |
| 103 def __init__(prob, name): | |
| 104 if not isinstance(name, basestring): | |
| 105 # This shouldn't happen, of course | |
| 21 | 106 raise TypeError('Problem() argument 1 must be string, not ' + type(name).__name__) |
| 16 | 107 prob.name = name |
| 21 | 108 prob.config = config.load_problem(name) |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
109 prob.cache = Cache({'padoutput': 0}) |
| 21 | 110 prob.testcases = testcases.load_problem(prob) |
| 111 | |
| 112 # TODO | |
| 113 def build(prob): | |
| 114 raise NotImplementedError | |
| 16 | 115 |
| 116 def test(prob): | |
| 23 | 117 case = None |
| 22 | 118 try: |
|
75
007f7eb6fb2b
The test context stack is now a deque.
Oleg Oshmyan <chortos@inbox.lv>
parents:
74
diff
changeset
|
119 contexts = deque((TestGroup(),)) |
| 22 | 120 for case in prob.testcases: |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
121 if case is test_context_end: |
|
76
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
122 real, max, log = contexts.pop().end() |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
123 for case, correct in log: |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
124 contexts[-1].case_start(case) |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
125 if correct: |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
126 contexts[-1].case_correct() |
|
76
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
127 contexts[-1].case_end() |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
128 contexts[-1].score(real, max) |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
129 continue |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
130 elif isinstance(case, TestContext): |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
131 contexts.append(case) |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
132 continue |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
133 contexts[-1].case_start(case) |
| 22 | 134 granted = 0 |
| 135 id = str(case.id) | |
| 136 if case.isdummy: | |
| 137 id = 'sample ' + id | |
| 138 say('%*s: ' % (prob.cache.padoutput, id), end='') | |
| 139 sys.stdout.flush() | |
| 140 try: | |
| 141 granted = case(lambda: (say('%7.3f%s s, ' % (case.time_stopped - case.time_started, case.time_limit_string), end=''), sys.stdout.flush())) | |
| 142 except testcases.CanceledByUser: | |
| 143 verdict = 'canceled by the user' | |
| 144 except testcases.TimeLimitExceeded: | |
| 145 verdict = 'time limit exceeded' | |
|
77
69eadc60f4e2
Memory limit is now applied to the RSS when os.wait4 is available
Oleg Oshmyan <chortos@inbox.lv>
parents:
76
diff
changeset
|
146 except testcases.MemoryLimitExceeded: |
|
69eadc60f4e2
Memory limit is now applied to the RSS when os.wait4 is available
Oleg Oshmyan <chortos@inbox.lv>
parents:
76
diff
changeset
|
147 verdict = 'memory limit exceeded' |
| 22 | 148 except testcases.WrongAnswer: |
| 149 e = sys.exc_info()[1] | |
| 150 if e.comment: | |
| 151 verdict = 'wrong answer (%s)' % e.comment | |
| 152 else: | |
| 153 verdict = 'wrong answer' | |
| 154 except testcases.NonZeroExitCode: | |
| 155 e = sys.exc_info()[1] | |
| 156 if e.exitcode < 0: | |
| 157 if sys.platform == 'win32': | |
| 158 verdict = 'terminated with error 0x%X' % (e.exitcode + 0x100000000) | |
| 159 elif -e.exitcode in signalnames: | |
| 160 verdict = 'terminated by signal %d (%s)' % (-e.exitcode, signalnames[-e.exitcode]) | |
| 161 else: | |
| 162 verdict = 'terminated by signal %d' % -e.exitcode | |
| 21 | 163 else: |
| 22 | 164 verdict = 'non-zero return code %d' % e.exitcode |
| 165 except testcases.CannotStartTestee: | |
| 26 | 166 verdict = 'cannot launch the program to test%s' % strerror(sys.exc_info()[1].upstream) |
| 22 | 167 except testcases.CannotStartValidator: |
| 26 | 168 verdict = 'cannot launch the validator%s' % strerror(sys.exc_info()[1].upstream) |
| 22 | 169 except testcases.CannotReadOutputFile: |
| 26 | 170 verdict = 'cannot read the output file%s' % strerror(sys.exc_info()[1].upstream) |
| 22 | 171 except testcases.CannotReadInputFile: |
| 26 | 172 verdict = 'cannot read the input file%s' % strerror(sys.exc_info()[1].upstream) |
| 22 | 173 except testcases.CannotReadAnswerFile: |
| 26 | 174 verdict = 'cannot read the reference output file%s' % strerror(sys.exc_info()[1].upstream) |
| 22 | 175 except testcases.TestCaseNotPassed: |
| 26 | 176 verdict = 'unspecified reason [this may be a bug in test.py]%s' % strerror(sys.exc_info()[1]) |
| 22 | 177 #except Exception: |
| 26 | 178 # verdict = 'unknown error [this may be a bug in test.py]%s' % strerror(sys.exc_info()[1]) |
| 21 | 179 else: |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
180 try: |
| 22 | 181 granted, comment = granted |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
182 except TypeError: |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
183 comment = '' |
|
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
184 else: |
| 22 | 185 if comment: |
| 186 comment = ' (%s)' % comment | |
|
24
c23d81f4a1a3
Score returned by TestCase.__call__() is now normalized to 0..1
Oleg Oshmyan <chortos@inbox.lv>
parents:
23
diff
changeset
|
187 if granted >= 1: |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
188 contexts[-1].case_correct() |
| 22 | 189 verdict = 'OK' + comment |
| 190 elif not granted: | |
| 191 verdict = 'wrong answer' + comment | |
| 192 else: | |
| 193 verdict = 'partly correct' + comment | |
| 26 | 194 granted *= case.points |
| 22 | 195 say('%g/%g, %s' % (granted, case.points, verdict)) |
|
76
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
196 contexts[-1].case_end() |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
197 contexts[-1].score(granted, case.points) |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
198 weighted = contexts[0].real * prob.config.taskweight / contexts[0].max if contexts[0].max else 0 |
|
76
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
199 before_weighting = valued = '' |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
200 if prob.config.taskweight != contexts[0].max: |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
201 before_weighting = ' (%g/%g before weighting)' % (contexts[0].real, contexts[0].max) |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
202 if contexts[0].nvalued != contexts[0].ntotal: |
|
76
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
203 valued = ' (%d/%d valued)' % (contexts[0].ncorrectvalued, contexts[0].nvalued) |
|
0e5ae28e0b2b
Points are now weighted on a test context basis
Oleg Oshmyan <chortos@inbox.lv>
parents:
75
diff
changeset
|
204 say('Problem total: %d/%d tests%s, %g/%g points%s' % (contexts[0].ncorrect, contexts[0].ntotal, valued, weighted, prob.config.taskweight, before_weighting)) |
|
39
2b459f9743b4
Test groups are now supported
Oleg Oshmyan <chortos@inbox.lv>
parents:
26
diff
changeset
|
205 sys.stdout.flush() |
| 22 | 206 return weighted, prob.config.taskweight |
| 207 finally: | |
| 23 | 208 if options.erase and (not prob.config.stdio or case and case.validator): |
| 22 | 209 for var in 'in', 'out': |
| 210 name = getattr(prob.config, var + 'name') | |
| 211 if name: | |
| 212 try: | |
| 213 os.remove(name) | |
| 214 except Exception: | |
| 215 pass | |
| 216 if case.validator and not callable(case.validator): | |
| 217 if prob.config.ansname: | |
| 218 try: | |
| 219 os.remove(prob.config.ansname) | |
| 220 except Exception: | |
| 221 pass |