def testGetWordForLanguage(self): L = Lexicon() L.addLanguage('en') L.addLanguage('de') L.insertWords((u'fox', u'quick'), 'en') L.insertWords((u'fuchs', u'quick'), 'de') en_words = L.getWordsForLanguage('en') en_words.sort() de_words = L.getWordsForLanguage('de') de_words.sort() self.assertEqual(en_words, ['fox', 'quick']) self.assertEqual(de_words, ['fuchs', 'quick']) self.assertRaises(LexiconError, L.getWordsForLanguage, 'unsupported_language')
def setUp(self): self._lexicon = Lexicon()