logic.py tests

unicefuganda · Aug 28, 2016 · 4cfcaac · 4cfcaac
1 parent cd63052
commit 4cfcaac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/survey/tests/models/test_ea_model.py b/survey/tests/models/test_ea_model.py
@@ -48,7 +48,7 @@ def test_get_survey_openings(self):
         ea_new.locations.add(self.kampala)
         self.batch.open_for_location(self.kampala)
         self.assertTrue(self.survey.is_open_for(self.kampala))
-        self.assertEquals(83, ea_new.get_survey_openings(self.survey).values()[0]["batch_id"])
+        self.assertEquals(self.batch.pk, ea_new.get_survey_openings(self.survey).values()[0]["batch_id"])
 
     def test_open_batches(self):
         ea1=EnumerationArea.objects.create(name="new EA")