From 417e6340004f4c0fa36a1026f740174a820465f9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ivar=20H=C3=A4rnqvist?= <ivaha717@student.liu.se> Date: Fri, 24 Apr 2020 18:05:02 +0200 Subject: [PATCH] remove unused dependency map from SFG, simplify dependency tests --- test/test_depends.py | 33 +++++++-------------------------- 1 file changed, 7 insertions(+), 26 deletions(-) diff --git a/test/test_depends.py b/test/test_depends.py index bb8897d5..e2691105 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 -- GitLab