def setUp(self): self.data = { "DC_PEC": "import numpy as np", "DC_SOLUTION": ''' try: x = max([1, 2, 'a']) except TypeError as e: x = 'typeerror' except ValueError: x = 'valueerror' except (ZeroDivisionError, IOError) as e: x = e except : x = 'someerror' else : passed = True finally: print('done') ''', "DC_SCT": ''' import collections handlers = collections.OrderedDict() handlers['TypeError'] = lambda: test_object_after_expression('x') handlers['ValueError'] = lambda: test_object_after_expression('x') handlers['ZeroDivisionError'] = lambda: test_object_after_expression('x', context_vals = ['anerror']) handlers['IOError'] = lambda: test_object_after_expression('x', context_vals = ['anerror']) handlers['all'] = lambda: test_object_after_expression('x') test_try_except(index = 1, body = lambda: test_function("max"), handlers = handlers, orelse = lambda: test_object_after_expression('passed'), finalbody = lambda: test_function('print')) ''' } self.SCT_NO_LAM = helper.remove_lambdas(self.data['DC_SCT'])
def test_fun_step2_no_lam(self): self.data["DC_SCT"] = helper.remove_lambdas(self.data["DC_SCT"]) self.test_fun_step2()
def test_Fail_no_lam(self): self.data["DC_SCT"] = helper.remove_lambdas(self.data["DC_SCT"]) self.test_Fail()
def test_Pass_no_lam(self): self.data["DC_SCT"] = helper.remove_lambdas(self.data["DC_SCT"]) self.test_Pass()
def test_Pass_no_lam(self): self.data["DC_SCT"] = helper.remove_lambdas(self.data["DC_SCT"], with_args=True) self.test_Pass()
def test_pass_no_lam(self): self.data["DC_CODE"] = "[key + str(val) for key,val in x.items() if isinstance(key, str) if isinstance(val, str)]" self.data["DC_SCT"] = helper.remove_lambdas(self.data["DC_SCT"]) sct_payload = helper.run(self.data) self.assertTrue(sct_payload['correct'])
def test_pass_mix_lam(self): self.data["DC_SCT"] = helper.remove_lambdas(self.data["DC_SCT"], count=2) self.test_pass()
def test_pass_no_lam(self): self.data["DC_CODE"] = "[[col for col in range(5)] for row in range(5)]" self.data["DC_SCT"] = helper.remove_lambdas(self.data["DC_SCT"]) sct_payload = helper.run(self.data) self.assertTrue(sct_payload['correct'])
def test_Pass2_mix_lam(self): self.data["DC_SCT"] = helper.remove_lambdas(self.data["DC_SCT"], count=1) self.test_Pass2()
def test_Fail2_no_lam(self): self.data["DC_SCT"] = helper.remove_lambdas(self.data["DC_SCT"], with_args=True) self.test_Fail2()
def test_Pass_exchain(self): self.data["DC_SCT"] = "Ex().\\" + helper.remove_lambdas(self.data["DC_SCT"]) self.test_Pass()