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