def test_extend(self): a = Partition([[1,2,3],[4,5],[8],[6,7]]) b = a.extend(3, [6]).extend(0, [1,2]) c = Partition([[3],[5,4],[8],[7],[6],[1,2]]) self.assertEqual(b,c)