def test_shots_execution_default(self, circuit, mocker):
"""Test correct tracks shots as well."""
class callback_wrapper:
@staticmethod
def callback(totals=dict(), history=dict(), latest=dict()):
pass
wrapper = callback_wrapper()
spy = mocker.spy(wrapper, "callback")
with Tracker(circuit.device, callback=wrapper.callback) as tracker:
circuit(shots=10)
circuit(shots=20)
assert tracker.totals == {"executions": 2, "shots": 30}
assert tracker.history == {"executions": [1, 1], "shots": [10, 20]}
assert tracker.latest == {"executions": 1, "shots": 20}
assert spy.call_count == 2
_, kwargs_called = spy.call_args_list[-1]
assert kwargs_called["totals"] == {"executions": 2, "shots": 30}
assert kwargs_called["history"] == {
"executions": [1, 1],
"shots": [10, 20]
}
assert kwargs_called["latest"] == {"executions": 1, "shots": 20}
def test_single_execution_default(self, circuit, mocker):
"""Test correct behavior with single circuit execution"""
class callback_wrapper:
@staticmethod
def callback(totals=dict(), history=dict(), latest=dict()):
pass
wrapper = callback_wrapper()
spy = mocker.spy(wrapper, "callback")
with Tracker(circuit.device, callback=wrapper.callback) as tracker:
circuit()
circuit()
assert tracker.totals == {"executions": 2}
assert tracker.history == {"executions": [1, 1], "shots": [None, None]}
assert tracker.latest == {"executions": 1, "shots": None}
_, kwargs_called = spy.call_args_list[-1]
assert kwargs_called["totals"] == {"executions": 2}
assert kwargs_called["history"] == {
"executions": [1, 1],
"shots": [None, None]
}
assert kwargs_called["latest"] == {"executions": 1, "shots": None}
def test_context(self):
"""Assert works with runtime context"""
with Tracker() as tracker:
assert isinstance(tracker, Tracker)
assert tracker.active == True
assert tracker.active == False
def test_incompatible_device_assignment_explicit_false(self):
"""Assert exception raised when `supports_tracker` is False"""
class TempDevice:
short_name = "temp"
temp = TempDevice()
with pytest.raises(
Exception,
match=r"Device 'temp' does not support device tracking"):
Tracker(dev=temp)
def test_device_assignment(self):
"""Assert gets assigned to device"""
class TempDevice:
short_name = "temp"
def __init__(self):
self.tracker = Tracker()
temp = TempDevice()
tracker = Tracker(dev=temp)
assert id(temp.tracker) == id(tracker)
def test_default_initialization(self):
"""Tests default initializalition"""
tracker = Tracker()
assert tracker.persistent == False
assert tracker.callback is None
assert tracker.history == dict()
assert tracker.totals == dict()
assert tracker.latest == dict()
assert tracker.active == False
def test_incompatible_device_assignment_no_tracker(self):
"""Assert exception raised when `supports_tracker` not True"""
class TempDevice:
short_name = "temp"
def capabilities(self):
return dict()
temp = TempDevice()
with pytest.raises(
Exception,
match=r"Device 'temp' does not support device tracking"):
Tracker(dev=temp)
def test_reset(self):
"""Assert reset empties totals, history and latest"""
tracker = Tracker()
tracker.totals = {"a": 1}
tracker.history = {"a": [1]}
tracker.latest = {"a": 1}
tracker.reset()
assert tracker.totals == dict()
assert tracker.history == dict()
assert tracker.latest == dict()
def test_enter_and_exit(self):
"""Assert entering and exit work as expected"""
tracker = Tracker()
tracker.totals = {"a": 1}
tracker.history = {"a": [1]}
tracker.latest = {"a": 1}
returned = tracker.__enter__()
assert id(tracker) == id(returned)
assert tracker.active == True
assert tracker.totals == dict()
assert tracker.history == dict()
assert tracker.latest == dict()
tracker.__exit__(None, None, None)
assert tracker.active == False
def test_batch_execution(self, circuit, mocker):
"""Tests that batch execute also updates information stored."""
class callback_wrapper:
@staticmethod
def callback(totals=dict(), history=dict(), latest=dict()):
pass
wrapper = callback_wrapper()
spy = mocker.spy(wrapper, "callback")
# initial execution to get qtape
circuit()
with Tracker(circuit.device, callback=wrapper.callback) as tracker:
circuit.device.batch_execute([circuit.qtape, circuit.qtape])
assert tracker.totals == {
"executions": 2,
"batches": 1,
"batch_len": 2
}
assert tracker.history == {
"executions": [1, 1],
"shots": [None, None],
"batches": [1],
"batch_len": [2],
}
assert tracker.latest == {"batches": 1, "batch_len": 2}
_, kwargs_called = spy.call_args_list[-1]
assert kwargs_called["totals"] == {
"executions": 2,
"batches": 1,
"batch_len": 2
}
assert kwargs_called["history"] == {
"executions": [1, 1],
"shots": [None, None],
"batches": [1],
"batch_len": [2],
}
assert kwargs_called["latest"] == {"batches": 1, "batch_len": 2}
def test_update(self):
"""Checks update stores to history and totals"""
tracker = Tracker()
tracker.update(a=1, b="b", c=None)
tracker.update(a=2, b="b2", c=1)
assert tracker.history == {
"a": [1, 2],
"b": ["b", "b2"],
"c": [None, 1]
}
assert tracker.totals == {"a": 3, "c": 1}
assert tracker.latest == {"a": 2, "b": "b2", "c": 1}
def test_record_callback(self, mocker):
class callback_wrapper:
@staticmethod
def callback(totals, history, latest):
pass
wrapper = callback_wrapper()
spy = mocker.spy(wrapper, "callback")
tracker = Tracker(callback=wrapper.callback)
tracker.totals = {"a": 1, "b": 2}
tracker.history = {"a": [1], "b": [1, 1]}
tracker.latest = {"a": 1, "b": 1}
tracker.record()
_, kwargs_called = spy.call_args_list[-1]
assert kwargs_called["totals"] == tracker.totals
assert kwargs_called["history"] == tracker.history
assert kwargs_called["latest"] == tracker.latest
def __init__(self):
self.tracker = Tracker()
