From fc34673e84ea9bc823e8530c7564f0caa6a99000 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Martin=20Mo=C5=BEina?= Date: Tue, 27 Sep 2016 16:38:23 +0200 Subject: Added exercises with list comprehensions. --- .../problems/comprehensions/sumsquares/common.py | 70 ++++++++++++++++++++++ 1 file changed, 70 insertions(+) create mode 100644 python/problems/comprehensions/sumsquares/common.py (limited to 'python/problems/comprehensions/sumsquares/common.py') diff --git a/python/problems/comprehensions/sumsquares/common.py b/python/problems/comprehensions/sumsquares/common.py new file mode 100644 index 0000000..bda561e --- /dev/null +++ b/python/problems/comprehensions/sumsquares/common.py @@ -0,0 +1,70 @@ +import re +from python.util import has_token_sequence, string_almost_equal, \ + string_contains_number, get_tokens, get_numbers, get_exception_desc, all_tokens,\ + get_ast, has_comprehension, has_loop +from server.hints import Hint + +id = 20701 +number = 1 +visible = True + +solution = '''\ +def sumsquares(n): + return sum(i**2 for i in range(n+1)) +''' + +hint_type = { + 'final_hint': Hint('final_hint'), + 'no_comprehension': Hint('no_comprehension') +} + +def test(python, code): + func_name = 'sumsquares' + tokens = get_tokens(code) + ast = get_ast(code) + if not has_token_sequence(tokens, ['def', func_name]): + return False, [{'id' : 'no_func_name', 'args' : {'func_name' : func_name}}] + + in_out = [ + (100, 338350), + (10, 385), + ] + + test_in = [('{0}({1})'.format(func_name, str(l[0])), None) + for l in in_out] + test_out = [l[1] for l in in_out] + + answers = python(code=code, inputs=test_in, timeout=1.0) + n_correct = 0 + tin, tout = None, None + for i, (ans, to) in enumerate(zip(answers, test_out)): + corr = ans[0] == to + n_correct += corr + if not corr: + tin = test_in[i][0] + tout = to + + passed = n_correct == len(test_in) + hints = [{'id': 'test_results', 'args': {'passed': n_correct, 'total': len(test_in)}}] + if tin: + hints.append({'id': 'problematic_test_case', 'args': {'testin': str(tin), 'testout': str(tout)}}) + if passed: + if not has_comprehension(ast): + hints.append({'id': 'no_comprehension'}) + elif has_loop(ast): + hints.append({'id': 'has_loop'}) + else: + hints.append({'id': 'final_hint'}) + + return passed, hints + + +def hint(python, code): + tokens = get_tokens(code) + + # run one test first to see if there are any exceptions + answer = python(code=code, inputs=[(None, None)], timeout=1.0) + exc = get_exception_desc(answer[0][3]) + if exc: return exc + + return None -- cgit v1.2.1