def items(self):
focus_area_page_params = []
for year in FinancialYear.get_available_years():
treemap = get_consolidated_expenditure_treemap(year)
for data in treemap["data"]["items"]:
focus_area_page_params.append({
"year": str(year.slug),
"focus": data["id"]
})
return focus_area_page_params
def test_complete_data(self, mock_get_dataset):
mock_get_dataset.return_value = self.mock_dataset
result = get_consolidated_expenditure_treemap(self.year)
data = result['data']
self.assertEqual(len(data), 2)
data_keys = data.keys()
self.assertIn('items', data_keys)
self.assertIn('total', data_keys)
expenditure_keys = data['items'][0].keys()
self.assertIn('name', expenditure_keys)
self.assertIn('amount', expenditure_keys)
self.assertIn('percentage', expenditure_keys)
self.assertIn('id', expenditure_keys)
self.assertIn('url', expenditure_keys)
def test_complete_data(self, mock_get_dataset):
mock_get_dataset.return_value = self.mock_dataset
result = get_consolidated_expenditure_treemap(self.year)
data = result["data"]
self.assertEqual(len(data), 2)
data_keys = data.keys()
self.assertIn("items", data_keys)
self.assertIn("total", data_keys)
expenditure_keys = data["items"][0].keys()
self.assertIn("name", expenditure_keys)
self.assertIn("amount", expenditure_keys)
self.assertIn("percentage", expenditure_keys)
self.assertIn("id", expenditure_keys)
self.assertIn("url", expenditure_keys)