def test_format_dictlist(self): dictlist = [ { 'label1': 'value-1,1', 'label2': 'value-1,2', 'label3': 'value-1,3', 'label4': 'value-1,4', }, { 'label1': 'value-2,1', 'label2': 'value-2,2', 'label3': 'value-2,3', 'label4': 'value-2,4', } ] features = ['label1', 'label4'] header, datarows = format_dictlist(dictlist, features) ideal_header = ['label1', 'label4'] ideal_datarows = [['value-1,1', 'value-1,4'], ['value-2,1', 'value-2,4']] self.assertEqual(header, ideal_header) self.assertEqual(datarows, ideal_datarows)
def test_format_dictlist(self): dictlist = [{ 'label1': 'value-1,1', 'label2': 'value-1,2', 'label3': 'value-1,3', 'label4': 'value-1,4', }, { 'label1': 'value-2,1', 'label2': 'value-2,2', 'label3': 'value-2,3', 'label4': 'value-2,4', }] features = ['label1', 'label4'] header, datarows = format_dictlist(dictlist, features) ideal_header = ['label1', 'label4'] ideal_datarows = [['value-1,1', 'value-1,4'], ['value-2,1', 'value-2,4']] self.assertEqual(header, ideal_header) self.assertEqual(datarows, ideal_datarows)
def test_format_dictlist_empty(self): header, datarows = format_dictlist([], []) self.assertEqual(header, []) self.assertEqual(datarows, [])