def test_partition_stabaliser(self): fam = PartitionStabaliserFamily(Partition([[1,2],[3,4]])) a = PartitionStack([0,0,0,0],[-1,-1,-1,-1]) func, _ = fam.extension_functions(a) func(a) funcs = fam.extension_functions(a) self.assertTrue(funcs is None) self.assertEqual(sorted(a[-1]), sorted(Partition([[1,2],[3,4]])))
def test_partition_stabaliser_with_none(self): stab = Partition([[1],[2],[3,4]]) fam = PartitionStabaliserFamily(stab) a = PartitionStack([0,0,0,0],[-1,-1,-1,-1]) b = PartitionStack([0,0,0,0],[-1,-1,-1,-1]) funcs1 = fam.extension_functions(a) funcs1[0](a) funcs2 = fam.extension_functions(a) funcs2[0](a) funcs3 = fam.extension_functions(a) self.assertTrue(funcs3 is None) funcs1[0](b) funcs2[0](b) self.assertEqual(a,b) self.assertEqual(len(a[-1]),3) self.assertEqual(sorted(stab), sorted(a[-1]))