def test_wordvectorizer_todataframe(test_text_info):
test_text, vocabulary = test_text_info
model = WordVectorizer().fit(test_text)
df = model.to_DataFrame()
if test_text == test_text_example:
assert df.shape == (7, 14)
else:
assert df.shape[0] <= len(vocabulary)
assert df.shape[1] == df.shape[0] * 2
def test_wordvectorizer_todataframe():
model = WordVectorizer().fit(test_text)
df = model.to_DataFrame()
assert df.shape == (7, 14)