class MembershipTest(unittest.TestCase): def setUp(self): self.membership = Membership('10', '03012020', 'silver') def test_getJoinDate(self): self.assertEqual(self.membership.getJoinDate(), '03012020') def test_setLevel(self): self.membership.setLevel('gold') self.assertEqual(self.membership.getLevel(), 'gold')