def test_batch_circuits_all_objects_circuit_set_is_artifact(
self, input_circuits, input_circuit_set_filename):
# Given
expected_circuit_set_filename = "circuit-set.json"
with open(input_circuit_set_filename) as f:
expected_circuitset = new_circuits.circuitset_from_dict(
json.load(f))
for circuit in input_circuits:
expected_circuitset.append(copy.deepcopy(circuit))
# When
batch_circuits(input_circuits, circuit_set=input_circuit_set_filename)
# Then
try:
with open(expected_circuit_set_filename) as f:
circuitset = new_circuits.circuitset_from_dict(json.load(f))
assert circuitset == expected_circuitset
finally:
remove_file_if_exists(expected_circuit_set_filename)
def test_batch_circuits_all_objects_no_circuit_set(self, input_circuits):
# Given
expected_circuit_set_filename = "circuit-set.json"
expected_circuitset = copy.deepcopy(input_circuits)
# When
batch_circuits(input_circuits)
# Then
try:
with open(expected_circuit_set_filename) as f:
circuitset = new_circuits.circuitset_from_dict(json.load(f))
assert circuitset == expected_circuitset
finally:
remove_file_if_exists(expected_circuit_set_filename)
def test_concatenate_circuits_artifact_file(self, circuit_set_filename):
# Given
expected_concatenated_circuit_filename = "result-circuit.json"
with open(circuit_set_filename) as f:
circuit_set = new_circuits.circuitset_from_dict(json.load(f))
expected_concatenated_circuit = sum(
[circuit for circuit in circuit_set], new_circuits.Circuit())
# When
concatenate_circuits(circuit_set_filename)
# Then
try:
with open(expected_concatenated_circuit_filename) as f:
concatenated_circuit = new_circuits.circuit_from_dict(
json.load(f))
assert concatenated_circuit == expected_concatenated_circuit
finally:
remove_file_if_exists(expected_concatenated_circuit_filename)
def test_batch_circuits_all_artifacts_no_circuit_set(
self, input_circuits_filenames):
# Given
expected_circuitset_filename = "circuit-set.json"
expected_circuitset = []
for circuit_filename in input_circuits_filenames:
with open(circuit_filename) as f:
circuit = new_circuits.circuit_from_dict(json.load(f))
expected_circuitset.append(circuit)
# When
batch_circuits(input_circuits_filenames)
# Then
try:
with open(expected_circuitset_filename) as f:
circuitset = new_circuits.circuitset_from_dict(json.load(f))
assert circuitset == expected_circuitset
finally:
remove_file_if_exists(expected_circuitset_filename)