class TestVectorItem(unittest.TestCase):
""" Tests the VectorItem class.
"""
def setUp(self):
text = u"Comment Google classe les pages Internet"
c = Classifier(CleanTextUtil("french"))
c.add_text(text)
self.dictionary_db = c.dictionary_db
self.vi = VectorItem("googl", "1")
def tearDown(self):
rm_data_dir()
def test_printer(self):
""" Tests printer.
Create a new vector item with a word of the text.
print the vector.
"""
print(self.vi.printer(self.dictionary_db))
def test_word_info(self):
""" Tests word_info.
Create a new vector item with a word of the text.
Call word_info with the dictionary db:
1- Verify the result object is a WordInfo object.
"""
word_info = self.vi.word_info(self.dictionary_db)
self.assertIsInstance(word_info, WordInfo) # 1
def setUp(self):
text = u"Comment Google classe les pages Internet"
c = Classifier(CleanTextUtil("french"))
c.add_text(text)
self.dictionary_db = c.dictionary_db
self.vi = VectorItem("googl", "1")