From dab6f458594699316c1272e26dbe2cf7a2f388e2 Mon Sep 17 00:00:00 2001 From: Kevin <Kevin> Date: Sun, 3 May 2020 14:25:20 +0200 Subject: [PATCH] cleanup test case --- test/test_sfg.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_sfg.py b/test/test_sfg.py index 2023ffcf..885b0be8 100644 --- a/test/test_sfg.py +++ b/test/test_sfg.py @@ -268,7 +268,7 @@ class TestReplaceOperations: mad1 = MAD() _sfg = sfg.replace_operations(['add1', 'mul1'], mad1) - assert _sfg.find_by_id('mad1') is not None + assert 'mad1' in _sfg._components_by_id.keys() assert {add1, mul1} not in _sfg.operations def test_replace_neg_add_with_sub(self): @@ -282,7 +282,7 @@ class TestReplaceOperations: sub1 = Subtraction() _sfg = sfg.replace_operations(['add1', 'neg1'], sub1) - assert _sfg.find_by_id('sub1') is not None + assert 'sub1' in _sfg._components_by_id.keys() assert {add1, neg1} not in _sfg.operations def test_different_input_output_count(self, operation_tree): -- GitLab