Resolve "Specify internal input/output dependencies of an Operation"

Closes #74 (closed)

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