def test_filling_data_for_missing_week_periods(self): stub_doc_1 = {"_week_start_at": d(2013, 4, 1), "_count": 5} stub_doc_2 = {"_week_start_at": d(2013, 4, 15), "_count": 5} period_data = PeriodData([stub_doc_1, stub_doc_2], period="week") period_data.fill_missing_periods(d(2013, 4, 1), d(2013, 4, 16)) assert_that(period_data.data(), has_length(3))
def test_filling_data_for_missing_month_periods(self): stub_doc_1 = { "_month_start_at": d(2013, 4, 1), "_count": 5 } stub_doc_2 = { "_month_start_at": d(2013, 6, 1), "_count": 6 } period_data = PeriodData([stub_doc_1, stub_doc_2], period="month") period_data.fill_missing_periods(d(2013, 4, 1), d(2013, 6, 2)) assert_that(period_data.data(), has_length(3))
def test_filling_data_for_missing_week_periods(self): stub_doc_1 = { "_week_start_at": d(2013, 4, 1), "_count": 5 } stub_doc_2 = { "_week_start_at": d(2013, 4, 15), "_count": 5 } period_data = PeriodData([stub_doc_1, stub_doc_2], period="week") period_data.fill_missing_periods(d(2013, 4, 1), d(2013, 4, 16)) assert_that(period_data.data(), has_length(3))
def test_filling_data_for_missing_month_periods(self): stub_doc_1 = {"_month_start_at": d(2013, 4, 1), "_count": 5} stub_doc_2 = {"_month_start_at": d(2013, 6, 1), "_count": 6} period_data = PeriodData([stub_doc_1, stub_doc_2], period="month") period_data.fill_missing_periods(d(2013, 4, 1), d(2013, 6, 2)) assert_that(period_data.data(), has_length(3))