def save_membership_data_to_csv(self, file_name, membership):
""" Method to save membership data to CSV file
Arguments:
file_name -- file where will be saved the membership
membership -- membership objtect that will be saved in the CSV file
"""
row = [str(membership.get_code()), membership.get_name(), str(membership.get_discount())]
list_membership = []
list_membership.append(row)
CsvOperations.write_data(file_name, list_membership)
def test_write_data_in_csv(self):
""" Test that all data are wrote in the csv file"""
expected = [["uno", "dos", "tres"], ["cuatro", "cinco", "seis"]]
CsvOperations.write_data("test.csv", expected)
self.assertEqual(CsvOperations.read_data("test.csv"), expected)
