def _build_stats_entry_from_date_query(self, date_query):
time_dict = build_time_dict(date_query)
result = []
for year in sorted(time_dict.keys()):
for month in sorted(time_dict[year].keys()):
result.append(('{} {}'.format(self._get_month_name(month), year), time_dict[year][month]))
return result
def test_build_time_dict():
test_input = [{
'_id': {
'month': 12,
'year': 2016
},
'count': 10
}, {
'_id': {
'month': 1,
'year': 2017
},
'count': 8
}]
expected_result = {2016: {12: 10}, 2017: {1: 8}}
assert build_time_dict(test_input) == expected_result
def test_build_time_dict(self):
test_input = [{
'_id': {
'month': 12,
'year': 2016
},
'count': 10
}, {
'_id': {
'month': 1,
'year': 2017
},
'count': 8
}]
expected_result = {2016: {12: 10}, 2017: {1: 8}}
self.assertEqual(build_time_dict(test_input), expected_result)