from calculator import Count import unittest class testcount(unittest.TestCase): def setUp(self): print "test start" def test_mul(self): j=Count(4,6) self.assertEqual(j.mul(),24) def tearDown(self): print "testcase" if __name__=='__mian__': suite=unittest.TestSuite() suite.addTest(testcount("test_mul")) runner=unittest.TextTestRunner() runner.run(suite) def test_add(self): j=Count(2,1) add=j.add(); assert(add==3) def test_sub(self): k=Count(4,5); substr=k.sub(); # mytest=testcount(); # print mytest.test_add(); # print mytest.test_sub();
class Count: def __init__(self,a,b): self.a=int(a) self.b=int(b) def add(self): return self.a+self.b def sub(self): return self.b-self.a def mul(self): return self.b*self.a def div(self): return self.b%self.a