def test_protocol_group_exceptions():
exception_protocol = ExceptionProtocol("exception_protocol")
protocol_group = ProtocolGroup("protocol_group")
protocol_group.add_protocols(exception_protocol)
with tempfile.TemporaryDirectory() as directory:
with pytest.raises(RuntimeError):
protocol_group.execute(directory, ComputeResources())
def test_protocol_group_execution():
protocol_a = DummyInputOutputProtocol("protocol_a")
protocol_a.input_value = 1
protocol_b = DummyInputOutputProtocol("protocol_b")
protocol_b.input_value = ProtocolPath("output_value", protocol_a.id)
protocol_group = ProtocolGroup("protocol_group")
protocol_group.add_protocols(protocol_a, protocol_b)
with tempfile.TemporaryDirectory() as directory:
protocol_group.execute(directory, ComputeResources())
value_path = ProtocolPath("output_value", protocol_group.id, protocol_b.id)
final_value = protocol_group.get_value(value_path)
assert final_value == protocol_a.input_value