def test_c(self):
"""Test everything is moved forward by almost one quarter."""
dates, values = [datetime(2011, 4, 1), datetime(2011, 7, 1)], []
# values do not care for this test
expected_dates = [datetime(2011, 6, 30, 23, 59, 59),
datetime(2011, 9, 30, 23, 59, 59)]
data = DataForCumulativeGraph(dates, values)
self.assertEqual(expected_dates, data.move_forward(dates, 'quarter'))
def test_d(self):
"""Test everything is moved forward by almost one year."""
dates, values = [datetime(2011, 4, 1), datetime(2012, 4, 1)], []
# values do not care for this test
expected_dates = [datetime(2012, 3, 31, 23, 59, 59),
datetime(2013, 3, 31, 23, 59, 59)]
data = DataForCumulativeGraph(dates, values)
self.assertEqual(expected_dates, data.move_forward(dates, 'year'))
def test_a(self):
"""Test everything is moved forward by almost one day."""
dates, values = [datetime(2011, 4, 4), datetime(2011, 4, 5)], []
# values do not care for this test
expected_dates = [datetime(2011, 4, 4, 23, 59, 59),
datetime(2011, 4, 5, 23, 59, 59)]
data = DataForCumulativeGraph(dates, values)
self.assertEqual(expected_dates, data.move_forward(dates, 'day'))
