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)
