def test_get_category_data(self): month_data = Data.get_month_data(2000, 1) data = Data.get_category_data(month_data, 1) expects = [ "コンビニ", "スーパー", "立替分1" ] self._assert_list(data, expects)
def get(self, request, *args, **kwargs): # validation if "year" in request.GET and "month" in request.GET: year = request.GET.get("year") month = request.GET.get("month") if not is_valid_date(year, month): return HttpResponseBadRequest("parameter error") else: return HttpResponseBadRequest("parameter error") # 今月のデータ monthly_data = Data.get_month_data(int(year), int(month)) # ジャンルごとの支出 positive_categories_outgo = {} for c in Category.list(): if c.show_order >= 0: d = Data.get_category_data(monthly_data, c.pk) positive_categories_outgo[ c.name] = Data.get_outgo_sum(d) - Data.get_temp_sum(d) context = { 'categories_outgo': positive_categories_outgo, } return render(request, '_chart_container_data.html', context)
def test_get_category_data_empty(self): month_data = Data.get_month_data(1999, 1) data = Data.get_category_data(month_data, 1) self.assertEqual(data.count(), 0)
def test_get_category_data_nothing(self): month_data = Data.get_month_data(2000, 1) data = Data.get_category_data(month_data, 1000) self.assertEqual(data.count(), 0)