def test_project_train_fasttext(app, document_corpus, testdatadir):
pytest.importorskip("annif.backend.fasttext")
with app.app_context():
project = annif.project.get_project('fasttext-fi')
project.train(document_corpus)
assert testdatadir.join('projects/fasttext-fi/fasttext-model').exists()
assert testdatadir.join('projects/fasttext-fi/fasttext-model').size() > 0
def test_project_train_fasttext(registry, document_corpus, testdatadir):
pytest.importorskip("annif.backend.fasttext")
project = registry.get_project('fasttext-fi')
project.train(document_corpus)
assert testdatadir.join('projects/fasttext-fi/fasttext-model').exists()
assert testdatadir.join('projects/fasttext-fi/fasttext-model').size() > 0
def test_project_train_tfidf_nodocuments(registry, empty_corpus):
project = registry.get_project('tfidf-fi')
with pytest.raises(NotSupportedException) as excinfo:
project.train(empty_corpus)
assert 'Cannot train tfidf project with no documents' in str(excinfo.value)
def test_project_train_tfidf(registry, document_corpus, testdatadir):
project = registry.get_project('tfidf-fi')
project.train(document_corpus)
assert testdatadir.join('projects/tfidf-fi/tfidf-index').exists()
assert testdatadir.join('projects/tfidf-fi/tfidf-index').size() > 0
def test_project_train_tfidf_nodocuments(app, empty_corpus):
with app.app_context():
project = annif.project.get_project('tfidf-fi')
with pytest.raises(NotSupportedException) as excinfo:
project.train(empty_corpus)
assert 'Cannot train tfidf project with no documents' in str(excinfo.value)
def test_project_train_tfidf(app, document_corpus, testdatadir):
with app.app_context():
project = annif.project.get_project('tfidf-fi')
project.train(document_corpus)
assert testdatadir.join('projects/tfidf-fi/tfidf-index').exists()
assert testdatadir.join('projects/tfidf-fi/tfidf-index').size() > 0
