reuse from test.utils

gojomo · Sep 12, 2020 · e7ef8da · e7ef8da
1 parent bb947b3
commit e7ef8da
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/gensim/test/test_fasttext.py b/gensim/test/test_fasttext.py
@@ -18,7 +18,8 @@
 from gensim.models.word2vec import LineSentence
 from gensim.models.fasttext import FastText as FT_gensim, FastTextKeyedVectors, _unpack
 from gensim.models.keyedvectors import KeyedVectors
-from gensim.test.utils import datapath, get_tmpfile, temporary_file, common_texts as sentences
+from gensim.test.utils import datapath, get_tmpfile, temporary_file, common_texts as sentences, \
+    lee_corpus_list as list_corpus
 from gensim.test.test_word2vec import TestWord2VecModel
 import gensim.models._fasttext_bin
 from gensim.models.fasttext_inner import compute_ngrams, compute_ngrams_bytes, ft_hash_bytes
@@ -44,15 +45,6 @@
 FT_CMD = os.path.join(FT_HOME, "fasttext") if FT_HOME else None
 
 
-class LeeCorpus(object):
-    def __iter__(self):
-        with open(datapath('lee_background.cor')) as f:
-            for line in f:
-                yield utils.simple_preprocess(line)
-
-
-list_corpus = list(LeeCorpus())
-
 new_sentences = [
     ['computer', 'artificial', 'intelligence'],
     ['artificial', 'trees'],