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