def test_move_to_end(self): bag = self.bag_type('aaabbc') bag.move_to_end('c') assert FrozenOrderedBag(bag) == FrozenOrderedBag('aaabbc') bag.move_to_end('a') assert FrozenOrderedBag(bag) == FrozenOrderedBag('bbcaaa') bag.move_to_end('c', last=False) assert FrozenOrderedBag(bag) == FrozenOrderedBag('cbbaaa') with cute_testing.RaiseAssertor(KeyError): bag.move_to_end('x') with cute_testing.RaiseAssertor(KeyError): bag.move_to_end('x', last=False)
def test_sort(self): bag = self.bag_type('aaabbc') bag.sort() assert FrozenOrderedBag(bag) == FrozenOrderedBag('aaabbc') bag.sort(key='cba'.index) assert FrozenOrderedBag(bag) == FrozenOrderedBag('cbbaaa')