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