Modifico tests para python3 y cambio assert de igualdad de resultados

datosgobar · Jun 13, 2017 · 346f1a5 · 346f1a5
1 parent 8ab3f5c
commit 346f1a5
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/tests/test_text_classifier.py b/tests/test_text_classifier.py
@@ -32,7 +32,7 @@ def test_get_similar(self):
                 "El edificio más antiguo tiene muchas cuadros caros porque era de un multimillonario",
                 "El edificio más moderno tiene muchas programadoras que comen manzanas durante el almuerzo grupal"
             ],
-            ids=map(str, range(4))
+            ids=list(map(str, range(4)))
         )
 
         ids, distancias, palabras_comunes = tc.get_similar(
@@ -42,10 +42,14 @@ def test_get_similar(self):
 
         self.assertEqual(ids, ['0', '3', '2', '1'])
         self.assertEqual(
-            palabras_comunes,
+            [
+                sorted(palabras)
+                for palabras in palabras_comunes
+            ]
+            ,
             [
                 [u'edificio', u'manzanas'],
-                [u'edificio', u'muchas', u'manzanas'],
+                [u'edificio', u'manzanas', u'muchas'],
                 [u'edificio', u'muchas'], [u'muchas']
             ]
         )
@@ -60,13 +64,13 @@ def test_classify(self):
                 "Para hacer una torta de naranja se necesita harina, huevos, leche, ralladura de naranja y polvo de hornear",
                 "Para hacer un lemon pie se necesita crema, ralladura de limón, huevos, leche y harina"
             ],
-            ids=map(str, range(6))
+            ids=list(map(str, range(6)))
         )
 
         # entrena un clasificador
         tc.make_classifier(
             name="recetas_classifier",
-            ids=map(str, range(6)),
+            ids=list(map(str, range(6))),
             labels=["Comida", "Comida", "Trago", "Trago", "Postre", "Postre"]
         )