def test_label_count_vectorizer(): vect = LabelCountVectorizer() d2l_mat = vect.transform(docs, labels) assert_array_equal( d2l_mat.todense(), np.asarray([[0, 2, 1, 0], [1, 0, 0, 2], [0, 1, 0, 1], [1, 0, 3, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 3, 0, 0]], dtype=np.int64)) assert_equal(vect.index2label_, {0: ir, 1: ml, 2: nlp, 3: tm})
def test_label_count_vectorizer(): vect = LabelCountVectorizer() d2l_mat = vect.transform(docs, labels) assert_array_equal(d2l_mat.todense(), np.asarray([[0, 2, 1, 0], [1, 0, 0, 2], [0, 1, 0, 1], [1, 0, 3, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 3, 0, 0]], dtype=np.int64)) assert_equal(vect.index2label_, {0: ir, 1: ml, 2: nlp, 3: tm})