From 2744ef01c40d127b24d67da2e4e65d6863fced02 Mon Sep 17 00:00:00 2001 From: Danny Hermes Date: Mon, 14 Nov 2016 15:16:58 -0800 Subject: [PATCH] Accounting extra data in language system test. The API has started returning metadata, e.g. {u'mid': u'/m/020bg'} and has added "painter" as a mention of Michelangelo. --- system_tests/language.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/system_tests/language.py b/system_tests/language.py index 5965392d4f5f..1e81ca12afc1 100644 --- a/system_tests/language.py +++ b/system_tests/language.py @@ -74,10 +74,11 @@ def _check_analyze_entities_result(self, entities): self.assertEqual(entity1.name, self.NAME1) self.assertEqual(entity1.entity_type, EntityType.PERSON) self.assertGreater(entity1.salience, 0.0) - self.assertEqual(entity1.mentions, [entity1.name]) + # Other mentions may occur, e.g. "painter". + self.assertIn(entity1.name, entity1.mentions) self.assertEqual(entity1.wikipedia_url, 'http://en.wikipedia.org/wiki/Caravaggio') - self.assertEqual(entity1.metadata, {}) + self.assertIsInstance(entity1.metadata, dict) # Verify entity 2. self.assertEqual(entity2.name, self.NAME2) self.assertEqual(entity2.entity_type, EntityType.LOCATION) @@ -85,7 +86,7 @@ def _check_analyze_entities_result(self, entities): self.assertEqual(entity2.mentions, [entity2.name]) self.assertEqual(entity2.wikipedia_url, 'http://en.wikipedia.org/wiki/Italy') - self.assertEqual(entity2.metadata, {}) + self.assertIsInstance(entity2.metadata, dict) # Verify entity 3. self.assertEqual(entity3.name, self.NAME3) choices = (EntityType.EVENT, EntityType.WORK_OF_ART) @@ -95,7 +96,7 @@ def _check_analyze_entities_result(self, entities): wiki_url = ('http://en.wikipedia.org/wiki/' 'The_Calling_of_St_Matthew_(Caravaggio)') self.assertEqual(entity3.wikipedia_url, wiki_url) - self.assertEqual(entity3.metadata, {}) + self.assertIsInstance(entity3.metadata, dict) def test_analyze_entities(self): document = Config.CLIENT.document_from_text(self.TEXT_CONTENT)