class SetListTests(TestCase):

    def setUp(self):
        self.setlist = SetList()

    def assertSetList(self, setlist, positions):
        self.assertEqual(
            len(setlist), len(positions),
            'len(%r) [%d] != len(%r) [%d]' % (
                setlist, len(setlist), positions, len(positions)))
        for index, (setlist_position, position) in enumerate(zip(setlist, positions)):
            self.assertEqual(
                set(setlist_position), set(position),
                'got[%d] [%r] != expected[%d] [%r]' % (
                    index, setlist_position, index, position))

    def assertSetListFlat(self, setlist, theset):
        self.assertEqual(set(setlist.flat), set(theset))

    def test_01_add_to_set(self):
        self.setlist[3].add('item1')
        self.assertSetList(self.setlist, [[], [], [], ['item1']])
        self.assertSetListFlat(self.setlist, ['item1'])

    def test_02_truncate(self):
        self.assertEqual(set(self.setlist[3]), set())
        self.assertSetList(self.setlist, [[], [], [], []])
        self.setlist.truncate()
        self.assertSetList(self.setlist, [])

    def test_03_discard_from_set(self):
        self.setlist[3].add('item1')
        self.setlist[3].discard('item1')
        self.assertSetListFlat(self.setlist, [])
        self.assertSetList(self.setlist, [])

    def test_04_assign_set(self):
        self.setlist[2] = ['item1', 'item2']
        self.assertSetList(self.setlist, [[], [], ['item1', 'item2']])

    def test_05_append_set(self):
        self.setlist[0] = []
        self.setlist.append(['item1', 'item2'])
        self.assertSetList(self.setlist, [[], ['item1', 'item2']])

    def test_05_initialize(self):
        data = [['item1'], [], ['item2', 'item3']]
        setlist = SetList(data)
        self.assertSetList(setlist, data)
Ejemplo n.º 2
0
    def lookup_ordered(self, normalized_spellings, create_missing=False):
        """
        Returns a 2-tuple of
         * a SetList which contains, for each word in
          `normalized_spellings`, a set of all meanings associated
           with the word
         * a set of all found words

        If `create_missing` is `True`, creates words and meanings for
        words not found.  In this case the set of found words contains
        all searched words.
        """
        Word._get_cache().seed(normalized_spellings)
        result = SetList()
        found_words = set()
        cache = {}
        for word in normalized_spellings:
            if word not in cache:
                meanings, words = self.lookup_splitting(
                    word, create_missing=create_missing)
                cache[word] = list(meanings)
                found_words.update(words)
            result.append(cache[word])
        return result, found_words
 def setUp(self):
     self.setlist = SetList()