1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
try:
from itertools import zip_longest
except ImportError:
from itertools import izip_longest as zip_longest
def prettify_exceptions(exception, message, traceback=None,
testcase=None, line_no=None):
err = {"type": "assertion",
"exception": exception,
"traceback": traceback,
"message": message
}
if exception == 'RuntimeError' or exception == 'RecursionError':
err["traceback"] = None
if exception == 'AssertionError':
value = ("Expected answer from the" +
" test case did not match the output")
if message:
err["message"] = message
else:
err["message"] = value
err["traceback"] = None
err["test_case"] = testcase
err["line_no"] = line_no
return err
def _get_incorrect_user_lines(exp_lines, user_lines):
err_line_numbers = []
for line_no, (expected_line, user_line) in \
enumerate(zip_longest(exp_lines, user_lines)):
if (not user_line or not expected_line or
user_line.strip() != expected_line.strip()):
err_line_numbers.append(line_no)
return err_line_numbers
def compare_outputs(expected_output, user_output, given_input=None):
given_lines = user_output.splitlines()
exp_lines = expected_output.splitlines()
msg = {"type": "stdio",
"given_input": given_input,
"expected_output": exp_lines,
"user_output": given_lines
}
ng = len(given_lines)
ne = len(exp_lines)
err_line_numbers = _get_incorrect_user_lines(exp_lines, given_lines)
msg["error_line_numbers"] = err_line_numbers
if ng != ne:
msg["error_msg"] = ("Incorrect Answer: "
+ "We had expected {} number of lines. ".format(ne)
+ "We got {} number of lines.".format(ng)
)
return False, msg
else:
if err_line_numbers:
msg["error_msg"] = ("Incorrect Answer: "
+ "Line number(s) {0} did not match."
.format(", ".join(
map(str, [x+1 for x in err_line_numbers])
)))
return False, msg
else:
msg["error_msg"] = "Correct Answer"
return True, msg
|
