class TestStudent(unittest.TestCase): @classmethod def setUpClass(cls): print('\nsetupClass\n') @classmethod def tearDownClass(cls): print('\nteardownClass') def setUp(self): print('\nsetUp') self.stu_1 = Student('Robin', 'Wills', '25000') self.stu_2 = Student('Mark', 'Smith', '28000') def tearDown(self): print('tearDown') def test_mail(self): print('test_mail') self.assertEqual(self.stu_1.mail, '*****@*****.**') self.assertEqual(self.stu_2.mail, '*****@*****.**') self.stu_1.first = 'Jenifer' self.stu_2.first = 'Graham' self.assertEqual(self.stu_1.mail, '*****@*****.**') self.assertEqual(self.stu_2.mail, '*****@*****.**') def test_fullname(self): print('test_fullname') self.assertEqual(self.stu_1.fullname, 'Robin Wills') self.assertEqual(self.stu_2.fullname, 'Mark Smith') self.stu_1.first = 'Jenifer' self.stu_2.first = 'Graham' self.assertEqual(self.stu_1.fullname, 'Jenifer Wills') self.assertEqual(self.stu_2.fullname, 'Graham Smith') def test_stipend_hike(self): print('test_stipend_hike') self.stu_1.apply_hike() self.stu_2.apply_hike() self.assertEqual(self.stu_1.stipend, 26250) self.assertEqual(self.stu_2.stipend, 29400) if __name__ == '__main__': unittest.main()
def test_stipend_hike(self): stu_1 = Student('Robin', 'Wills', 25000) stu_2 = Student('Mark', 'Smith', 28000) stu_1.apply_hike() stu_2.apply_hike() self.assertEqual(stu_1.stipend, 26250) self.assertEqual(stu_2.stipend, 29400)