def test_wordsList_validinput(self): from database import WordParser from database import WordRecord, AbilityRecord contents = " dummy foo123 _ bar_bar dummy foobar [1] [22] kkk [1 zzz" parser = WordParser(contents) words = parser.wordsList() self.assertIsInstance(words, list) self.assertEqual(words[0], WordRecord("dummy")) self.assertEqual(words[1], WordRecord("foo")) self.assertEqual(words[2], WordRecord("bar")) self.assertEqual(words[3], WordRecord("bar")) self.assertEqual(words[4], WordRecord("dummy")) self.assertEqual(words[5], WordRecord("foobar")) self.assertEqual(words[6], AbilityRecord("[1]")) self.assertEqual(words[7], AbilityRecord("[22]")) self.assertEqual(words[8], WordRecord("kkk")) self.assertIn(WordRecord("kkk"), words) self.assertNotIn(WordRecord("[1"), words) self.assertNotIn(WordRecord("foo123"), words) self.assertNotIn(WordRecord("bar_bar"), words) self.assertNotIn(WordRecord(""), words)
def test_wordsList_invalidinput(self): from database import WordParser, WordParserException parser = WordParser(None) with self.assertRaises(WordParserException): words = parser.wordsList()