import unittest
class Method:
def add(self, a, b):
return a+b
def mul(self, a, b):
return a*b
class Test(unittest.TestCase):
a=Method()
def test_add(self):
self.assertEqual(3, self.a.add(1,2))
def test_mul(self):
self.assertEqual(3, self.a.mul(1,2))
unittest.main()