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