From 311f41f5df246880e4062eed83ec2463b0f24563 Mon Sep 17 00:00:00 2001 From: Kevin <Kevin> Date: Mon, 4 May 2020 17:38:18 +0200 Subject: [PATCH] Fixed sfg being immutable when replacing operations --- b_asic/signal_flow_graph.py | 2 +- test/test_sfg.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/b_asic/signal_flow_graph.py b/b_asic/signal_flow_graph.py index d449a0a7..d2089f17 100644 --- a/b_asic/signal_flow_graph.py +++ b/b_asic/signal_flow_graph.py @@ -444,7 +444,7 @@ class SFG(AbstractOperation): _signal.remove_source() _signal.set_source(operation.output(index_out)) - return _sfg + return _sfg() def _evaluate_source(self, src: OutputPort, results: MutableResultMap, registers: MutableRegisterMap, prefix: str) -> Number: src_prefix = prefix diff --git a/test/test_sfg.py b/test/test_sfg.py index 885b0be8..72737fda 100644 --- a/test/test_sfg.py +++ b/test/test_sfg.py @@ -274,13 +274,13 @@ class TestReplaceOperations: def test_replace_neg_add_with_sub(self): in1 = Input() in2 = Input() - neg1 = ConstantMultiplication(-1, in1, 'neg1') + neg1 = ConstantMultiplication(-1, in1) add1 = neg1 + in2 out1 = Output(add1) sfg = SFG(inputs=[in1, in2], outputs=[out1]) sub1 = Subtraction() - _sfg = sfg.replace_operations(['add1', 'neg1'], sub1) + _sfg = sfg.replace_operations(['add1', 'cmul1'], sub1) assert 'sub1' in _sfg._components_by_id.keys() assert {add1, neg1} not in _sfg.operations -- GitLab