def test_sort_natural(self):
periods = [
MonthPeriod('201607'),
MonthPeriod('201601'),
MonthPeriod('201501'),
MonthPeriod('201507'),
]
periods.sort()
self.assertEqual([p.period for p in periods], ['201501', '201507', '201601', '201607'])
def test_current_past_future(self):
month = MonthPeriod()
self.assertTrue(month.is_current())
self.assertFalse(month.next().is_current())
self.assertTrue(month.next().is_future())
self.assertFalse(month.next().is_past())
self.assertTrue(month.previous().is_past())
self.assertFalse(month.next().is_past())
def test_empty_constructor_uses_current_month(self):
current_year = date.today().year
current_month = date.today().month
period = MonthPeriod()
self.assertEqual(period.year, current_year)
self.assertEqual(period.month, current_month)
def test_create_from_year_only_uses_january_as_month(self):
self.assertEqual(MonthPeriod(year=2012), MonthPeriod('201201'))
def test_create_from_empty_strings_creates_for_current_month(self):
self.assertEqual(str(MonthPeriod(yearmonth='')), MonthPeriod())
self.assertEqual(str(MonthPeriod(year='', month='')), MonthPeriod())
def test_create_from_strings(self):
self.assertEqual(str(MonthPeriod(year='2012', month='02')), '201202')
def test_range_with_invalid_bounds_yield_nothing(self):
start = MonthPeriod('201512')
generator = start.range(2, -2)
self.assertRaises(StopIteration, next, generator)
def test_range_negative(self):
start = MonthPeriod('201512')
generator = start.range(-2, 2)
self.assertEqual([str(m) for m in generator], ['201510', '201511', '201512', '201601', '201602'])
def test_deserialize_from_string(self):
self.assertEqual(_deserialize('2013'), YearPeriod('2013'))
self.assertEqual(_deserialize('201305'), MonthPeriod('201305'))
def test_period_factory(self):
self.assertEqual(period_factory('year', '2015'), YearPeriod(2015))
self.assertEqual(period_factory('month', '201512'), MonthPeriod(year=2015, month=12))
def test_next(self):
self.assertEqual(MonthPeriod('201611').next(), MonthPeriod('201612'))
def test_add(self):
self.assertEqual(MonthPeriod('201611') + 1, MonthPeriod('201612'))
self.assertEqual(MonthPeriod('201611') + 2, MonthPeriod('201701'))
self.assertEqual(MonthPeriod('201602') + 14, MonthPeriod('201704'))
def test_subtract(self):
self.assertEqual(MonthPeriod('201602') - 1, MonthPeriod('201601'))
self.assertEqual(MonthPeriod('201602') - 2, MonthPeriod('201512'))
self.assertEqual(MonthPeriod('201602') - 14, MonthPeriod('201412'))
def test_format_long(self):
period = MonthPeriod('201612')
self.assertEqual(period.format_long(), 'December 2016')
def test_month_year_extraction(self):
period = MonthPeriod('201604')
self.assertEqual(period.year, 2016)
self.assertEqual(period.month, 4)
def test_serialize(self):
self.assertEqual(_serialize(YearPeriod('2012')), {'year': 2012})
self.assertEqual(_serialize(MonthPeriod('201202')), {'year': 2012, 'month': 2})
def test_previous(self):
self.assertEqual(MonthPeriod('201611').previous(), MonthPeriod('201610'))
def test_range(self):
start = MonthPeriod('201510')
generator = start.range(0, 3)
self.assertEqual([str(m) for m in generator], ['201510', '201511', '201512', '201601'])
def test_from_string_month(self):
self.assertEqual(period_from_string('2015'), YearPeriod(2015))
self.assertEqual(period_from_string('201512'), MonthPeriod(year=2015, month=12))
def test_deserialize_from_object(self):
self.assertEqual(_deserialize({'year': 2012}), YearPeriod('2012'))
self.assertEqual(_deserialize({'year': 2012, 'month': 2}), MonthPeriod('201202'))