Update parse tests

weni-ai · Aug 10, 2018 · 9787731 · 9787731
1 parent ff516e8
commit 9787731
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/bothub_nlp/core/tests/test_parse.py b/bothub_nlp/core/tests/test_parse.py
@@ -53,11 +53,5 @@ def test_parse(self):
         response = parse_text(self.update, example.get('text'))
 
         entities = response.get('entities')
-        print('entities', len(entities))
-        for entity in entities:
-            print(entity.get('value'), entity.get('entity'))
-
         labels_as_entity = response.get('labels_as_entity', [])
-        print('labels_as_entity', len(labels_as_entity))
-        for label_as_entity in labels_as_entity:
-            print(label_as_entity.get('value'), label_as_entity.get('entity'))
+        self.assertEqual(len(entities), len(labels_as_entity))