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