def test_partition(self): p = Partition('abcde') self.assertEqual([['a'], ['b'], ['c'], ['d'], ['e']], p.sets()) p.merge('a', 'b') self.assertEqual([['a', 'b'], ['c'], ['d'], ['e']], p.sets()) p.merge('c', 'd') self.assertEqual([['a', 'b'], ['c', 'd'], ['e']], p.sets()) p.merge('a', 'd') self.assertEqual([['a', 'c', 'b', 'd'], ['e']], p.sets()) p.merge('b', 'e') self.assertEqual([['a', 'c', 'b', 'e', 'd']], p.sets())