def test_19_truncate_comma_list(self): r = truncate_comma_list("123456,234567,345678", 19) self.assertEqual(len(r), 19) self.assertEqual(r, "1234+,234567,345678") r = truncate_comma_list("123456,234567,345678", 18) self.assertEqual(len(r), 18) self.assertEqual(r, "1234+,2345+,345678") r = truncate_comma_list("123456,234567,345678", 16) self.assertEqual(len(r), 16) self.assertEqual(r, "123+,2345+,3456+") # There are more entries than the max_len. We will not be able # to shorten all entries, so we simply take the beginning of the string. r = truncate_comma_list("12,234567,3456,989,123,234,234", 4) self.assertEqual(len(r), 4) self.assertEqual(r, "12,+")
def _truncate_data(self): """ Truncate self.audit_data according to the self.custom_column_length. :return: None """ for column, l in self.custom_column_length.items(): if column in self.audit_data: data = self.audit_data[column] if isinstance(data, string_types): if column == "policies": # The policies column is shortend per comma entry data = truncate_comma_list(data, l) else: data = data[:l] self.audit_data[column] = data
def _truncate_data(self): """ Truncate self.audit_data according to the column_length. :return: None """ for column, l in column_length.items(): if column in self.audit_data: data = self.audit_data[column] if isinstance(data, string_types): if column == "policies": # The policies column is shortend per comma entry data = truncate_comma_list(data, l) else: data = data[:l] self.audit_data[column] = data