def test_word(self):
     word = Word("test")
     self.assertEqual(word.getName(), "test")
     word.setTF(0.4)
     word.setIDF(0.6)
     self.assertEqual(word.getTF(), 0.4)
     self.assertEqual(word.getIDF(), 0.6)
 def test_document(self):
     word = Word("test")
     word.setTF(0.4)
     word.setIDF(0.6)
     doc = Document("hello my name is")
     self.assertEqual(doc.getDocString(), "hello my name is")
     doc.addRealWord(word)
     self.assertEqual(doc.getWordDict().get("test").getName(), "test")
     self.assertTrue(doc.hasWord("test"))