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)