Skip to content

Commit

Permalink
Merge pull request #3299 from pabs3/path-to-python-binary
Browse files Browse the repository at this point in the history
Enable test_word2vec_stand_alone_script by using sys.executable for python
  • Loading branch information
piskvorky authored Apr 15, 2022
2 parents 317e447 + 3ad9841 commit de9ee81
Showing 1 changed file with 14 additions and 9 deletions.
23 changes: 14 additions & 9 deletions gensim/test/test_word2vec.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
import bz2
import sys
import tempfile
import subprocess

import numpy as np

Expand All @@ -27,6 +28,7 @@

from gensim import utils
from gensim.models import word2vec, keyedvectors
from gensim.utils import check_output
from gensim.test.utils import (
datapath, get_tmpfile, temporary_file, common_texts as sentences,
LeeCorpus, lee_corpus_list,
Expand Down Expand Up @@ -1168,15 +1170,18 @@ def test_path_line_sentences_one_file(self):

# endclass TestWord2VecSentenceIterators

# TODO: get correct path to Python binary
# class TestWord2VecScripts(unittest.TestCase):
# def test_word2vec_stand_alone_script(self):
# """Does Word2Vec script launch standalone?"""
# cmd = 'python -m gensim.scripts.word2vec_standalone -train ' + datapath('testcorpus.txt') + \
# ' -output vec.txt -size 200 -sample 1e-4 -binary 0 -iter 3 -min_count 1'
# output = check_output(cmd, stderr=PIPE)
# self.assertEqual(output, '0')
# #endclass TestWord2VecScripts

class TestWord2VecScripts(unittest.TestCase):
def test_word2vec_stand_alone_script(self):
"""Does Word2Vec script launch standalone?"""
cmd = [
sys.executable, '-m', 'gensim.scripts.word2vec_standalone',
'-train', datapath('testcorpus.txt'),
'-output', 'vec.txt', '-size', '200', '-sample', '1e-4',
'-binary', '0', '-iter', '3', '-min_count', '1',
]
output = check_output(args=cmd, stderr=subprocess.PIPE)
self.assertEqual(output, b'')


if not hasattr(TestWord2VecModel, 'assertLess'):
Expand Down

0 comments on commit de9ee81

Please sign in to comment.