def test_empty_qs(self): qs = [] gf = GapFiller(qs, WEEKLY, self.date_key, DATE_FORMAT_WEEKLY) result = gf.fill_gaps() expected = [] self.assertEqual(result, expected)
def test_multiple_params(self): qs = [{ "date": "2015-01", "a": 1, "b": 1 }, { "date": "2015-03", "a": 3, "b": 4 }] gf = GapFiller(qs, MONTHLY, self.date_key, DATE_FORMAT_MONTHLY) result = gf.fill_gaps() expected = [ { "date": "2015-01", "a": 1, "b": 1 }, { "date": "2015-02", "a": 0, "b": 0 }, { "date": "2015-03", "a": 3, "b": 4 }, ] self.assertEqual(result, expected)
def test_no_gaps(self): qs = [ { "date": "2015-01", "param": 1 }, { "date": "2015-02", "param": 0 }, { "date": "2015-03", "param": 3 }, ] gf = GapFiller(qs, WEEKLY, self.date_key, DATE_FORMAT_WEEKLY) result = gf.fill_gaps() expected = [ { "date": "2015-01", "param": 1 }, { "date": "2015-02", "param": 0 }, { "date": "2015-03", "param": 3 }, ] self.assertEqual(result, expected)
def test_one_element(self): qs = [{"date": "2015-01", "param": 1}] gf = GapFiller(qs, MONTHLY, self.date_key, DATE_FORMAT_MONTHLY) result = gf.fill_gaps() expected = [{"date": "2015-01", "param": 1}] self.assertEqual(result, expected)
def test_one_element(self): qs = [ {'date': "2015-01", 'param': 1} ] gf = GapFiller(qs, MONTHLY, self.date_key, DATE_FORMAT_MONTHLY) result = gf.fill_gaps() expected = [ {'date': "2015-01", 'param': 1} ] self.assertEqual(result, expected)
def test_multiple_month_gaps(self): qs = [ { "date": "2015-01", "param": 1 }, { "date": "2015-03", "param": 3 }, { "date": "2015-07", "param": 7 }, ] gf = GapFiller(qs, MONTHLY, self.date_key, DATE_FORMAT_MONTHLY) result = gf.fill_gaps() expected = [ { "date": "2015-01", "param": 1 }, { "date": "2015-02", "param": 0 }, { "date": "2015-03", "param": 3 }, { "date": "2015-04", "param": 0 }, { "date": "2015-05", "param": 0 }, { "date": "2015-06", "param": 0 }, { "date": "2015-07", "param": 7 }, ] self.assertEqual(result, expected)
def test_multiple_params(self): qs = [ {'date': "2015-01", 'a': 1, 'b': 1}, {'date': "2015-03", 'a': 3, 'b': 4} ] gf = GapFiller(qs, MONTHLY, self.date_key, DATE_FORMAT_MONTHLY) result = gf.fill_gaps() expected = [ {'date': "2015-01", 'a': 1, 'b': 1}, {'date': "2015-02", 'a': 0, 'b': 0}, {'date': "2015-03", 'a': 3, 'b': 4} ] self.assertEqual(result, expected)
def test_single_week_gap(self): qs = [ {'date': "2015-01", 'param': 1}, {'date': "2015-03", 'param': 3} ] gf = GapFiller(qs, WEEKLY, self.date_key, DATE_FORMAT_WEEKLY) result = gf.fill_gaps() expected = [ {'date': "2015-01", 'param': 1}, {'date': "2015-02", 'param': 0}, {'date': "2015-03", 'param': 3} ] self.assertEqual(result, expected)
def test_multiple_month_gaps(self): qs = [ {'date': "2015-01", 'param': 1}, {'date': "2015-03", 'param': 3}, {'date': "2015-07", 'param': 7} ] gf = GapFiller(qs, MONTHLY, self.date_key, DATE_FORMAT_MONTHLY) result = gf.fill_gaps() expected = [ {'date': "2015-01", 'param': 1}, {'date': "2015-02", 'param': 0}, {'date': "2015-03", 'param': 3}, {'date': "2015-04", 'param': 0}, {'date': "2015-05", 'param': 0}, {'date': "2015-06", 'param': 0}, {'date': "2015-07", 'param': 7} ] self.assertEqual(result, expected)