From 43146e2ecb55dc8434b4a194417f12af4de54253 Mon Sep 17 00:00:00 2001 From: Clemens Wolff Date: Mon, 6 Jan 2020 15:33:54 -0500 Subject: [PATCH] Simplify test for statistics API --- app/api/tests/test_api.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/app/api/tests/test_api.py b/app/api/tests/test_api.py index 2c8b66bf..f806c222 100644 --- a/app/api/tests/test_api.py +++ b/app/api/tests/test_api.py @@ -1332,10 +1332,8 @@ class TestStatisticsAPI(APITestCase): self.client.login(username=self.super_user_name, password=self.super_user_pass) response = self.client.get(self.url, format='json') - total = self.doc.count() - remaining = self.doc.filter(doc_annotations__isnull=True).count() - self.assertEqual(response.data['total'], total) - self.assertEqual(response.data['remaining'], remaining) + self.assertEqual(response.data['total'], 2) + self.assertEqual(response.data['remaining'], 1) def test_returns_user_count(self): self.client.login(username=self.super_user_name,