Fixes piskvorky#1401 , tests for phrasified sentences added

sj29-innovate · Jan 31, 2018 · 3d26f32 · 3d26f32
1 parent dfdd16e
commit 3d26f32
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/gensim/test/test_phrases.py b/gensim/test/test_phrases.py
@@ -165,7 +165,10 @@ def testEmptyPhrasifiedSentencesIterator(self):
         bigram_phraser = Phraser(bigram_phrases)
         trigram_phrases = Phrases(bigram_phraser[self.sentences])
         trigram_phraser = Phraser(trigram_phrases)
-        self.assertNotEqual(trigram_phraser[bigram_phraser[self.sentences]].__len__(), 0)
+        trigrams = trigram_phraser[bigram_phraser[self.sentences]]
+        fst, snd = list(trigrams), list(trigrams)
+        self.assertEqual(fst, snd)
+        self.assertNotEqual(snd, [])
 
     def testEmptyInputsOnBigramConstruction(self):
         """Test that empty inputs don't throw errors and return the expected result."""