def test_get_info_none(self):
"""Test that get_info returns None if there is no active queuing context"""
A = qml.RZ(0.5, wires=1)
with AnnotatedQueue() as q:
q.append(A, inv=True)
assert QueuingContext.get_info(A) is None
def test_update_info(self):
"""Test that update_info correctly updates an annotation"""
A = qml.RZ(0.5, wires=1)
with AnnotatedQueue() as q:
q.append(A, inv=True)
assert QueuingContext.get_info(A) == {"inv": True}
assert q._get_info(A) == {"inv": True}
q._update_info(A, inv=False, owner=None)
assert q._get_info(A) == {"inv": False, "owner": None}