class TestMainHandler(unittest.TestCase):
def setUp(self):
self.my_object = MainHandler()
def test_add_company(self):
self.my_object.add_company('nitansh')
self.assertEqual(1 , len(self.my_object.company_list))
def test_read_from_file(self):
self.my_object.read_from_file('sample.csv')
self.assertEqual(5, len(self.my_object.company_list))
def setUp(self):
self.my_object = MainHandler()
