def test_wordsSet(self): from database import WordParser from database import WordRecord contents = "dummy foo123 bar_bar dummy foobar" parser = WordParser(contents) words = parser.wordsSet() self.assertIsInstance(words, set) self.assertIn(WordRecord("dummy"), words) self.assertIn(WordRecord("foobar"), words) self.assertNotIn(WordRecord("foo123"), words) self.assertNotIn(WordRecord("bar_bar"), words) self.assertNotIn(WordRecord(""), words)
def test_wordsSet_invalidinput(self): from database import WordParser, WordParserException parser = WordParser(None) with self.assertRaises(WordParserException): words = parser.wordsSet()