diff --git a/test/test_depends.py b/test/test_depends.py
index bb8897d53453d5ac967371586f2a626e0d8c01ef..e26911054a9604db2f08998f6ecfccd81a012e5a 100644
--- a/test/test_depends.py
+++ b/test/test_depends.py
@@ -3,36 +3,17 @@ from b_asic import Addition, Butterfly
 class TestDepends:
     def test_depends_addition(self):
         add1 = Addition()
-        out1_dependencies = add1.inputs_required_for_output(0)
-        assert len(list(out1_dependencies)) == 2
-        assert 0 in out1_dependencies
-        assert 1 in out1_dependencies
+        assert set(add1.inputs_required_for_output(0)) == {0, 1}
 
     def test_depends_butterfly(self):
         bfly1 = Butterfly()
-        out1_dependencies = bfly1.inputs_required_for_output(0)
-        out2_dependencies = bfly1.inputs_required_for_output(1)
-        assert len(list(out1_dependencies)) == 2
-        assert 0 in out1_dependencies
-        assert 1 in out1_dependencies
-        assert len(list(out2_dependencies)) == 2
-        assert 0 in out2_dependencies
-        assert 1 in out2_dependencies
+        assert set(bfly1.inputs_required_for_output(0)) == {0, 1}
+        assert set(bfly1.inputs_required_for_output(1)) == {0, 1}
 
     def test_depends_sfg(self, sfg_two_inputs_two_outputs):
-        out1_dependencies = sfg_two_inputs_two_outputs.inputs_required_for_output(0)
-        out2_dependencies = sfg_two_inputs_two_outputs.inputs_required_for_output(1)
-        assert len(list(out1_dependencies)) == 2
-        assert 0 in out1_dependencies
-        assert 1 in out1_dependencies
-        assert len(list(out2_dependencies)) == 2
-        assert 0 in out2_dependencies
-        assert 1 in out2_dependencies
+        assert set(sfg_two_inputs_two_outputs.inputs_required_for_output(0)) == {0, 1}
+        assert set(sfg_two_inputs_two_outputs.inputs_required_for_output(1)) == {0, 1}
 
     def test_depends_sfg_independent(self, sfg_two_inputs_two_outputs_independent):
-        out1_dependencies = sfg_two_inputs_two_outputs_independent.inputs_required_for_output(0)
-        out2_dependencies = sfg_two_inputs_two_outputs_independent.inputs_required_for_output(1)
-        assert len(list(out1_dependencies)) == 1
-        assert 0 in out1_dependencies
-        assert len(list(out2_dependencies)) == 1
-        assert 1 in out2_dependencies
\ No newline at end of file
+        assert set(sfg_two_inputs_two_outputs_independent.inputs_required_for_output(0)) == {0}
+        assert set(sfg_two_inputs_two_outputs_independent.inputs_required_for_output(1)) == {1}
\ No newline at end of file