def test_mask_parameter_attr(self): p = Parameter("masking_param", set_cmd=None, initial_value=1) p.x = 1 with Measurement("mask_parameter_attr") as msmt: self.assertEqual(p.x, 1) msmt.mask(p, x=2) self.assertEqual(len(msmt._masked_properties), 1) self.assertDictEqual( msmt._masked_properties[0], { "type": "attr", "obj": p, "attr": "x", "original_value": 1, "value": 2, }, ) for k in Sweep(range(10), "repetition"): msmt.measure({"a": 3, "b": 4}, "acquire_values") self.assertEqual(p.x, 2) self.assertEqual(p(), 1) self.assertEqual(p.x, 2) self.assertEqual(p.x, 1) self.assertEqual(len(msmt._masked_properties), 0)
def test_unmask_parameter_attr(self): p = Parameter("masking_param", set_cmd=None, initial_value=1) p.x = 1 with Measurement("mask_parameter") as msmt: msmt.mask(p, 2) msmt.mask(p, x=3) self.assertEqual(p(), 2) self.assertEqual(p.x, 3) msmt.unmask(p, attr="x") self.assertEqual(p(), 2) self.assertEqual(p.x, 1) self.assertEqual(p(), 1) self.assertEqual(p.x, 1)