def test_rename_nonexistent_name(self):
pipeline = CompVizPipeline()
with pytest.raises(ValueError) as e:
pipeline.rename_operator('nonexistent', 'foo')
assert_terms_in_exception(e, ['operator', 'name'])
def test_rename_keeps_operators_order(self, order_asserter_operator):
pipeline = CompVizPipeline()
pipeline.add_operator('op1', order_asserter_operator(0))
pipeline.add_operator('op2', order_asserter_operator(1))
pipeline.add_operator('op3', order_asserter_operator(2))
pipeline.rename_operator('op2', 'new_name')
pipeline.run(build_img((16, 16)))
def test_rename_keeps_order(self):
pipeline = CompVizPipeline()
pipeline.add_operator('op1', TestOperator())
pipeline.add_operator('op2', TestOperator())
pipeline.add_operator('op3', TestOperator())
pipeline.rename_operator('op2', 'renamed')
assert list(pipeline.operators.keys()) == ['op1', 'renamed', 'op3']
def test_rename_to_existent_name(self):
pipeline = CompVizPipeline()
pipeline.add_operator('op1', TestOperator())
pipeline.add_operator('op2', TestOperator())
with pytest.raises(ValueError) as e:
pipeline.rename_operator('op1', 'op2')
assert_terms_in_exception(e, ['name'])
def test_rename_operator(self):
pipeline = CompVizPipeline()
starting_name = 'some_name'
new_name = 'another_name'
op = TestOperator()
pipeline.add_operator(starting_name, op)
pipeline.add_operator('foo', TestOperator())
pipeline.add_operator('bar', TestOperator())
expected_len = len(pipeline.operators)
pipeline.rename_operator(starting_name, new_name)
assert len(pipeline.operators) == expected_len
assert new_name in pipeline.operators
assert pipeline.operators[new_name] is op
def test_rename_keeps_saving_order(self):
pipeline = CompVizPipeline()
pipeline.add_operator('op1', TestOperator())
pipeline.add_operator('op2', TestOperator())
pipeline.add_operator('op3', TestOperator())
pipeline.rename_operator('op2', 'renamed')
stream = StringIO()
pipeline.save(stream)
stream.seek(0)
config = yaml.safe_load(stream)
operators_order = [
op_config['name'] for op_config in config['pipeline']
]
assert operators_order == ['op1', 'renamed', 'op3']
def test_rename_to_same_name_ok(self):
op_name = 'some_op_name'
pipeline = CompVizPipeline()
pipeline.add_operator(op_name, TestOperator())
pipeline.rename_operator(op_name, op_name)
assert list(pipeline.operators.keys()) == [op_name]
