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')