Esempio n. 1
0
def test_multi_column_tfidf_vectorizer_vocabulary_sizes_error():
    with pytest.raises(ValueError):
        vectorizer = MultiColumnTfidfVectorizer(vocabulary_sizes=[1])
        vectorizer.fit(corpus)
Esempio n. 2
0
def test_multi_column_tfidf_vectorizer_transform_dim_error():
    with pytest.raises(ValueError):
        vec = MultiColumnTfidfVectorizer()
        vec.fit(corpus)
        vec.transform(corpus[0])