get_numero_escuela now returns a python set

2 files changed, 3 insertions, 3 deletions

diff --git a/webapp/polls/models.py b/webapp/polls/models.py
index 2e34a90..50f3ca9 100644
--- a/webapp/polls/models.py
+++ b/webapp/polls/models.py
@@ -1519,7 +1519,7 @@ class PollResultFile(object):
             self.data = json.loads(self.json_str, 'utf-8')
 
     def get_numero_escuela(self):
-        return self.data['result']['0']['polled']['NUM_ESC']
+        return set([self.data['result']['0']['polled']['NUM_ESC']])
 
     def get_pollster_username(self):
         return self.data['pollster_username']
diff --git a/webapp/polls/tests/poll_result_file_tests.py b/webapp/polls/tests/poll_result_file_tests.py
index 2ff63e8..b57e475 100644
--- a/webapp/polls/tests/poll_result_file_tests.py
+++ b/webapp/polls/tests/poll_result_file_tests.py
@@ -150,7 +150,7 @@ class NumeroEscuelaTest(MongoTestCase):
         data['result']['0']['polled'] = {}
         self.data = data
 
-    def test_it_should_respond_to_numero_escuela(self):
+    def test_it_should_respond_to_numero_escuela_as_a_set(self):
         data = self.data
         numero_escuela = '1'
         data['result']['0']['polled']['NUM_ESC'] = numero_escuela
@@ -163,4 +163,4 @@ class NumeroEscuelaTest(MongoTestCase):
         file_path = file_.name
 
         result = PollResultFile(file_path)
-        self.assertEqual(numero_escuela, result.get_numero_escuela())
+        self.assertEqual(set([numero_escuela]), result.get_numero_escuela())