def test_length(self): column1 = MonthsColumn(None, date(2012, 12, 18), date(2013, 7, 1)) self.assertEqual(len(column1), 8) column2 = MonthsColumn(None, date(2012, 12, 18), date(2012, 12, 19)) self.assertEqual(len(column2), 1) column3 = MonthsColumn(None, date(2012, 1, 18), date(2012, 12, 19)) self.assertEqual(len(column3), 12)
def test_headers(self): column1 = MonthsColumn(None, date(2012, 12, 18), date(2012, 12, 19)) self.assertEqual(column1.headers, [ 'December', ]) column2 = MonthsColumn(None, date(2012, 12, 18), date(2013, 2, 1)) self.assertEqual(column2.headers, [ 'December', 'January', 'February', ])