Exemplo n.º 1
0
 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)
Exemplo n.º 2
0
 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',
     ])