Fix flaky movie_nl tests

Jon Wayne Parrott · Jon Wayne Parrott · commit 0a86e80ac711 · 2017-08-16T09:51:07.000-07:00
Change-Id: I4922637173048627f38b507588a4f30a5d490212
diff --git a/language/movie_nl/main_test.py b/language/movie_nl/main_test.py
@@ -70,18 +70,18 @@ def test_process_movie_reviews():
     entities = [json.loads(entity) for entity in entities]
 
     # assert sentiments
-    assert sentiments[0].get('sentiment') == 0.9
+    assert sentiments[0].get('sentiment') > 0
     assert sentiments[0].get('label') == 1
 
-    assert sentiments[1].get('sentiment') == 0.9
+    assert sentiments[1].get('sentiment') > 0
     assert sentiments[1].get('label') == 1
 
     # assert entities
     assert len(entities) == 1
     assert entities[0].get('name') == 'Tom Cruise'
     assert (entities[0].get('wiki_url') ==
-            'http://en.wikipedia.org/wiki/Tom_Cruise')
-    assert entities[0].get('sentiment') == 1.8
+            'https://en.wikipedia.org/wiki/Tom_Cruise')
+    assert entities[0].get('sentiment') > 0
 
 
 def test_rank_positive_entities(capsys):
