class TestCompanyInfoFunctions(unittest.TestCase):
def setUp(self):
self.my_object = CompanyInfo('nitansh')
def test_add_info(self):
self.my_object.add_info('2010', 'feb', '100')
local_string = self.my_object.share_info()
self.assertEqual('nitansh ---> the peak time is 2010 feb and Share at that time is 100 ', local_string)
def test_share_peak_info(self):
self.my_object.add_info('2010', 'feb', '150')
self.my_object.add_info('2011', 'feb', '200')
self.assertEqual(('2011 feb ', '200'), self.my_object.find_peak_share_time())
self.my_object.add_info('2011', 'jan', '200')
self.assertEqual(('2011 feb 2011 jan ', '200'), self.my_object.find_peak_share_time())
def setUp(self):
self.my_object = CompanyInfo('nitansh')
