Moving assertion outside of a context manager (piskvorky#2611)

lopusz · Jan 22, 2020 · 6009f0d · 6009f0d
1 parent 7eef237
commit 6009f0d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/gensim/test/test_fasttext.py b/gensim/test/test_fasttext.py
@@ -1432,7 +1432,10 @@ def _check_roundtrip_file_file(self, sg):
             _save_test_model(m1_basename, model_params)
             model = gensim.models.fasttext.load_facebook_model(m1)
             gensim.models.fasttext.save_facebook_model(model, m2)
-            self.assertEqual(_read_binary_file(m1), _read_binary_file(m2))
+            bin1 = _read_binary_file(m1)
+            bin2 = _read_binary_file(m2)
+
+        self.assertEqual(bin1, bin2)
 
     def test_skipgram(self):
         self._check_roundtrip_file_file(sg=1)