def test_discount_is_correctly_saved_in_csv(self):
""" Method to test if the memnershipd discount is correctly saved in the csv file"""
manage_membership = ManageMembership()
manage_membership.save_membership_data_to_csv(self.file_name, self.membership)
retrieved_data = manage_membership.read_membership_data_from_csv(self.file_name)
self.assertEqual(10, int(retrieved_data[2]))
def test_membership_name_is_correctly_saved_in_csv(self):
""" Method to test if the memnershipd name is correctly saved in the csv file"""
manage_membership = ManageMembership()
manage_membership.save_membership_data_to_csv(self.file_name, self.membership)
retrieved_data = manage_membership.read_membership_data_from_csv(self.file_name)
self.assertEqual("Gold", retrieved_data[1])
