def test__add_columns(self): column_set = ReportColumnSet() self.assertEqual(len(column_set), 0) column_set.add_columns(sim_columns()) self.assertEqual(len(column_set), 2) for x in column_set.report_columns: self.assertTrue(isinstance(x, dict)) self.assertEqual(column_set.report_columns[0]['number'], 1) self.assertEqual(column_set.report_columns[1]['number'], 2) # Additional columns can be added, the number fields should be set # properly. column_set.add_columns(sim_columns()) self.assertEqual(len(column_set), 4) self.assertEqual(column_set.report_columns[2]['number'], 3) self.assertEqual(column_set.report_columns[3]['number'], 4)
def test__sorted(self): column_set = ReportColumnSet() self.assertEqual(column_set.sorted(), []) column_set.add_columns(sim_columns()) sorted_set = column_set.sorted() self.assertEqual(sorted_set[0]['heading'], 'ID') self.assertEqual(sorted_set[1]['heading'], 'Name') # Reverse the numbers column_set.report_columns[0]['number'] = 2 column_set.report_columns[1]['number'] = 1 sorted_set = column_set.sorted() self.assertEqual(sorted_set[0]['heading'], 'Name') self.assertEqual(sorted_set[1]['heading'], 'ID')