def test_query(self): checker = SpellChecker() validTestData = [('hello', 'Rare'), ('fantastic', 'Rare'), ('order', 'Common'), ('absolute', 'Uncommon'), ('smartphone', 'Unknown'), ('aksfjalsf', 'Unknown')] invalidTestData = [315, 0, -4] for query, response in validTestData: self.assertTrue(checker.query(query) == response) for invalidQuery in invalidTestData: with self.assertRaises(TypeError): checker.query(invalidQuery)