def setUp(self):
self.words = [['one', 'un', 'uno', 'ichi'],
['red', 'rouge', 'rojo'],
['good', 'great', 'awesome'],
]
self.lex = Lexicon(self.words)
class Test_Lexicon(unittest.TestCase):
def setUp(self):
self.words = [['one', 'un', 'uno', 'ichi'],
['red', 'rouge', 'rojo'],
['good', 'great', 'awesome'],
]
self.lex = Lexicon(self.words)
def test_init_with_words_adds_entry_for_each_word(self):
self.assertItemsEqual(self.lex['red'], ['red', 'rouge', 'rojo'])
def test_add_words_adds_entry_for_each_word(self):
words = ['blue', 'bleu', 'azul']
self.lex.add_words(words)
for word in words:
self.assertItemsEqual(self.lex[word], words)
def test_adding_overlapping_wordlist_merges_wordlists(self):
self.lex.add_words(['green', 'vert'])
self.lex.add_words(['vert', 'verde'])
self.assertItemsEqual(self.lex['verde'], ['green', 'vert', 'verde'])
def test_bridging_wordlists_merges_wordlists(self):
self.lex.add_words(['red', 'good'])
wordlist = ['red', 'rouge', 'rojo', 'good', 'great', 'awesome']
self.assertItemsEqual(self.lex['red'], wordlist)
self.assertItemsEqual(self.lex['good'], wordlist)
self.assertItemsEqual(self.lex['rojo'], wordlist)
self.assertItemsEqual(self.lex['great'], wordlist)
def test_word_match_returns_true_for_synonyms(self):
self.lex.add_words(['green', 'vert'])
self.assertTrue(self.lex.words_match('green', 'vert'))
self.assertTrue(self.lex.words_match('vert', 'green'))
def test_get_word_sets_returns_sets_of_synonyms(self):
self.assertItemsEqual(self.lex.get_word_sets(),
[set(words) for words in self.words])
def test_get_word_sets_returns_unique_sets_of_synonyms(self):
self.lex.add_words(['red', 'good'])
self.assertItemsEqual(self.lex.get_word_sets(),
[set(['one', 'un', 'uno', 'ichi']),
set(['red', 'rouge', 'rojo', 'good', 'great', 'awesome'])
])
