# -*- 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()