Skip to content

Commit

Permalink
Remove unnecessary test; add test for analyzing with passes > 1 setting
Browse files Browse the repository at this point in the history
  • Loading branch information
osma committed Jan 30, 2019
1 parent b251014 commit 7038111
Showing 1 changed file with 12 additions and 13 deletions.
25 changes: 12 additions & 13 deletions tests/test_backend_vw.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,19 +96,6 @@ def test_vw_train_invalid_learning_rate(datadir, project, vw_corpus):
vw.train(vw_corpus, project)


def test_vw_train_unknown_subject(datadir, project, vw_corpus):
vw_type = annif.backend.get_backend('vw_multi')
vw = vw_type(
backend_id='vw_multi',
params={'chunksize': 4},
datadir=str(datadir))

vw.train(vw_corpus, project)
assert vw._model is not None
assert datadir.join('vw-model').exists()
assert datadir.join('vw-model').size() > 0


def test_vw_analyze(datadir, project):
vw_type = annif.backend.get_backend('vw_multi')
vw = vw_type(
Expand Down Expand Up @@ -141,6 +128,18 @@ def test_vw_analyze_empty(datadir, project):
assert len(results) == 0


def test_vw_analyze_multiple_passes(datadir, project):
vw_type = annif.backend.get_backend('vw_multi')
vw = vw_type(
backend_id='vw_multi',
params={'chunksize': 4, 'passes': 2},
datadir=str(datadir))

results = vw.analyze("...", project)

assert len(results) == 0


def test_vw_train_ect(datadir, document_corpus, project):
vw_type = annif.backend.get_backend('vw_multi')
vw = vw_type(
Expand Down

0 comments on commit 7038111

Please sign in to comment.