# -*- coding: UTF-8 -*- import codity @codity.trace_entry_and_exit def solution(a, b): return a + b if __name__ == "__main__": c = codity.CodityRunner() c.run(1, 2) c.run(2, 2) c.add_vector(3, 1, 2) c.add_vector(3, 2, 2) c.run_test()
def test_3(self): cr = codity.CodityRunner(demo2) cr.add_vector(0, 0, 0) cr.add_vector(1, 0, 0) cr.run_test()
def test_2(self): cr = codity.CodityRunner(demo2) ret = cr.run(2, 2) self.assertEqual(ret, 0)
def test_1(self): cr = codity.CodityRunner(demo1) ret = cr.run(1) self.assertEqual(ret, 0)
def test_0(self): cr = codity.CodityRunner() self.assertEqual(cr.solution, None) cr = codity.CodityRunner(demo0) self.assertEqual(cr.solution, demo0)
import codity @codity.trace_entry_and_exit def asolution(A): raise IndexError() if __name__ == "__main__": c = codity.CodityRunner(asolution) c.run(1) c.add_vector(1, 1) c.add_vector(2, 1) c.run_test()