def test_get_survival_rate(self): result = get_survival_rate(self.day1, 2, 2) self.assertEqual(2, result['unit']) self.assertEqual(2, result['user_window']) self.assertEqual(week_start(self.day1), result['start_date']) self.assertEqual(4, result['total']) values = result['values'] self.assertEqual(4, values[0]['y']) self.assertEqual(2, values[1]['y']) self.assertEqual(1, values[2]['y']) self.assertEqual(1, values[3]['y']) self.assertEqual(1, values[4]['y']) for x in range(5, 10): self.assertEqual(0, values[x]['y'])
def get_survival_rate(request): request_date = __get_request_date(request) user_window = __get_request_interval_unit(request) user_window = int(user_window) result = survival.get_survival_rate(request_date, user_window, 2) return JsonResponse(result)