diff --git a/test/fixtures/port.py b/test/fixtures/port.py
index fa528b8d9437e60b99c1ec426f317eb97b0164f2..4cce4f69b1f11b44426d1bd39702dba4e11c0efe 100644
--- a/test/fixtures/port.py
+++ b/test/fixtures/port.py
@@ -10,3 +10,11 @@ def input_port():
@pytest.fixture
def output_port():
return OutputPort(None, 0)
+
+@pytest.fixture
+def list_of_input_ports():
+ return [InputPort(None, i) for i in range(0, 3)]
+
+@pytest.fixture
+def list_of_output_ports():
+ return [OutputPort(None, i) for i in range(0, 3)]
diff --git a/test/test_inputport.py b/test/test_inputport.py
index 85f892217c7e0f766417f6cc2e6d066d48d8a537..f8b9facc4b86599071dfcfd5df39ee3f8fb063ee 100644
--- a/test/test_inputport.py
+++ b/test/test_inputport.py
@@ -7,15 +7,7 @@ import pytest
from b_asic import InputPort, OutputPort, Signal
@pytest.fixture
-def inp_port():
- return InputPort(None, 0)
-
-@pytest.fixture
-def out_port():
- return OutputPort(None, 0)
-
-@pytest.fixture
-def out_port2():
+def output_port2():
return OutputPort(None, 1)
@pytest.fixture
@@ -23,53 +15,53 @@ def dangling_sig():
return Signal()
@pytest.fixture
-def s_w_source(out_port):
- return Signal(source=out_port)
+def s_w_source(output_port):
+ return Signal(source=output_port)
@pytest.fixture
def sig_with_dest(inp_port):
return Signal(destination=inp_port)
@pytest.fixture
-def connected_sig(inp_port, out_port):
- return Signal(source=out_port, destination=inp_port)
+def connected_sig(inp_port, output_port):
+ return Signal(source=output_port, destination=inp_port)
-def test_connect_then_disconnect(inp_port, out_port):
+def test_connect_then_disconnect(input_port, output_port):
"""Test connect unused port to port."""
- s1 = inp_port.connect(out_port)
+ s1 = input_port.connect(output_port)
- assert inp_port.connected_source == out_port
- assert inp_port.signals == [s1]
- assert out_port.signals == [s1]
- assert s1.source is out_port
- assert s1.destination is inp_port
+ assert input_port.connected_source == output_port
+ assert input_port.signals == [s1]
+ assert output_port.signals == [s1]
+ assert s1.source is output_port
+ assert s1.destination is input_port
- inp_port.remove_signal(s1)
+ input_port.remove_signal(s1)
- assert inp_port.connected_source is None
- assert inp_port.signals == []
- assert out_port.signals == [s1]
- assert s1.source is out_port
+ assert input_port.connected_source is None
+ assert input_port.signals == []
+ assert output_port.signals == [s1]
+ assert s1.source is output_port
assert s1.destination is None
-def test_connect_used_port_to_new_port(inp_port, out_port, out_port2):
+def test_connect_used_port_to_new_port(input_port, output_port, output_port2):
"""Does connecting multiple ports to an inputport throw error?"""
- inp_port.connect(out_port)
+ input_port.connect(output_port)
with pytest.raises(Exception):
- inp_port.connect(out_port2)
+ input_port.connect(output_port2)
-def test_add_signal_then_disconnect(inp_port, s_w_source):
+def test_add_signal_then_disconnect(input_port, s_w_source):
"""Can signal be connected then disconnected properly?"""
- inp_port.add_signal(s_w_source)
+ input_port.add_signal(s_w_source)
- assert inp_port.connected_source == s_w_source.source
- assert inp_port.signals == [s_w_source]
+ assert input_port.connected_source == s_w_source.source
+ assert input_port.signals == [s_w_source]
assert s_w_source.source.signals == [s_w_source]
- assert s_w_source.destination is inp_port
+ assert s_w_source.destination is input_port
- inp_port.remove_signal(s_w_source)
+ input_port.remove_signal(s_w_source)
- assert inp_port.connected_source is None
- assert inp_port.signals == []
+ assert input_port.connected_source is None
+ assert input_port.signals == []
assert s_w_source.source.signals == [s_w_source]
assert s_w_source.destination is None
diff --git a/test/test_outputport.py b/test/test_outputport.py
index 189c89225f88f263294d24aae21995ee7a821ada..723d18ff3ca8ac69d50af642d50b75a45a3ae1ec 100644
--- a/test/test_outputport.py
+++ b/test/test_outputport.py
@@ -5,20 +5,6 @@ import pytest
from b_asic import OutputPort, InputPort, Signal
-
-@pytest.fixture
-def output_port():
- return OutputPort(None, 0)
-
-@pytest.fixture
-def input_port():
- return InputPort(None, 0)
-
-@pytest.fixture
-def list_of_input_ports():
- return [InputPort(None, i) for i in range(0, 3)]
-
-
class TestConnect:
def test_multiple_ports(self, output_port, list_of_input_ports):
"""Can multiple ports connect to an output port?"""
@@ -43,7 +29,7 @@ class TestAddSignal:
assert output_port.signal_count == 1
assert output_port.signals == [s]
-class TestDisconnect:
+class TestClear:
def test_others_clear(self, output_port, list_of_input_ports):
"""Can multiple ports disconnect from OutputPort?"""
for port in list_of_input_ports:
diff --git a/test/test_print_sfg.py b/test/test_print_sfg.py
deleted file mode 100644
index e2e253aa6fc11f9f47c1c11fb8a68ab2baf409d4..0000000000000000000000000000000000000000
--- a/test/test_print_sfg.py
+++ /dev/null
@@ -1,53 +0,0 @@
-"""
-B-ASIC test suite for printing a SFG
-"""
-
-
-from b_asic.signal_flow_graph import SFG
-from b_asic.core_operations import Addition, Multiplication, Constant
-from b_asic.port import InputPort, OutputPort
-from b_asic.signal import Signal
-from b_asic.special_operations import Input, Output
-
-import pytest
-
-
-class TestPrintSfg:
- def test_print_one_addition(self):
- inp1 = Input("INP1")
- inp2 = Input("INP2")
- add1 = Addition(inp1, inp2, "ADD1")
- out1 = Output(add1, "OUT1")
- sfg = SFG(inputs=[inp1, inp2], outputs=[out1], name="sf1")
-
- assert sfg.__str__() == ("id: add1, name: ADD1, input: [s1, s2], output: [s3]\nid: in1, name: INP1, input: [], output: [s1]\nid: in2, name: INP2, input: [], output: [s2]\nid: out1, name: OUT1, input: [s3], output: []\n")
-
- def test_print_add_mul(self):
- inp1 = Input("INP1")
- inp2 = Input("INP2")
- inp3 = Input("INP3")
- add1 = Addition(inp1, inp2, "ADD1")
- mul1 = Multiplication(add1, inp3, "MUL1")
- out1 = Output(mul1, "OUT1")
- sfg = SFG(inputs=[inp1, inp2, inp3], outputs=[out1], name="mac_sfg")
-
- assert sfg.__str__() == ("id: add1, name: ADD1, input: [s1, s2], output: [s5]\nid: in1, name: INP1, input: [], output: [s1]\nid: in2, name: INP2, input: [], output: [s2]\nid: mul1, name: MUL1, input: [s5, s3], output: [s4]\nid: in3, name: INP3, input: [], output: [s3]\nid: out1, name: OUT1, input: [s4], output: []\n")
-
- def test_print_constant(self):
- inp1 = Input("INP1")
- const1 = Constant(3, "CONST")
- add1 = Addition(const1, inp1, "ADD1")
- out1 = Output(add1, "OUT1")
-
- sfg = SFG(inputs=[inp1], outputs=[out1], name="sfg")
-
- assert sfg.__str__() == ("id: add1, name: ADD1, input: [s3, s1], output: [s2]\nid: c1, name: CONST, value: 3, input: [], output: [s3]\nid: in1, name: INP1, input: [], output: [s1]\nid: out1, name: OUT1, input: [s2], output: []\n")
-
- def test_print_simple_filter(self, simple_filter):
- assert simple_filter.__str__() == \
- 'id: add1, name: , input: [s1, s3], output: [s4]\n' + \
- 'id: in1, name: , input: [], output: [s1]\n' + \
- 'id: cmul1, name: , input: [s5], output: [s3]\n' + \
- 'id: reg1, name: , input: [s4], output: [s5, s2]\n' + \
- 'id: out1, name: , input: [s2], output: []\n'
-
\ No newline at end of file
diff --git a/test/test_signal_flow_graph.py b/test/test_sfg.py
similarity index 69%
rename from test/test_signal_flow_graph.py
rename to test/test_sfg.py
index 51267cc44ece60c05e622c02c89b8ec1a5d5b17d..f41f9c4967192b7064fd637200516ab78a2c2414 100644
--- a/test/test_signal_flow_graph.py
+++ b/test/test_sfg.py
@@ -3,7 +3,7 @@ import pytest
from b_asic import SFG, Signal, Input, Output, Constant, Addition, Multiplication
-class TestConstructor:
+class TestInit:
def test_direct_input_to_output_sfg_construction(self):
in1 = Input("IN1")
out1 = Output(None, "OUT1")
@@ -45,21 +45,58 @@ class TestConstructor:
assert sfg.input_count == 0
assert sfg.output_count == 1
+class TestStr:
+ def test_one_addition(self):
+ inp1 = Input("INP1")
+ inp2 = Input("INP2")
+ add1 = Addition(inp1, inp2, "ADD1")
+ out1 = Output(add1, "OUT1")
+ sfg = SFG(inputs=[inp1, inp2], outputs=[out1], name="sf1")
-class TestEvaluation:
- def test_evaluate_output(self, operation_tree):
- sfg = SFG(outputs = [Output(operation_tree)])
- assert sfg.evaluate_output(0, []) == 5
+ assert sfg.__str__() == \
+ "id: add1, name: ADD1, input: [s1, s2], output: [s3]\n" + \
+ "id: in1, name: INP1, input: [], output: [s1]\n" + \
+ "id: in2, name: INP2, input: [], output: [s2]\n" + \
+ "id: out1, name: OUT1, input: [s3], output: []\n"
- def test_evaluate_output_large(self, large_operation_tree):
- sfg = SFG(outputs = [Output(large_operation_tree)])
- assert sfg.evaluate_output(0, []) == 14
+ def test_add_mul(self):
+ inp1 = Input("INP1")
+ inp2 = Input("INP2")
+ inp3 = Input("INP3")
+ add1 = Addition(inp1, inp2, "ADD1")
+ mul1 = Multiplication(add1, inp3, "MUL1")
+ out1 = Output(mul1, "OUT1")
+ sfg = SFG(inputs=[inp1, inp2, inp3], outputs=[out1], name="mac_sfg")
- def test_evaluate_output_cycle(self, operation_graph_with_cycle):
- sfg = SFG(outputs = [Output(operation_graph_with_cycle)])
- with pytest.raises(Exception):
- sfg.evaluate_output(0, [])
+ assert sfg.__str__() == \
+ "id: add1, name: ADD1, input: [s1, s2], output: [s5]\n" + \
+ "id: in1, name: INP1, input: [], output: [s1]\n" + \
+ "id: in2, name: INP2, input: [], output: [s2]\n" + \
+ "id: mul1, name: MUL1, input: [s5, s3], output: [s4]\n" + \
+ "id: in3, name: INP3, input: [], output: [s3]\n" + \
+ "id: out1, name: OUT1, input: [s4], output: []\n"
+ def test_constant(self):
+ inp1 = Input("INP1")
+ const1 = Constant(3, "CONST")
+ add1 = Addition(const1, inp1, "ADD1")
+ out1 = Output(add1, "OUT1")
+
+ sfg = SFG(inputs=[inp1], outputs=[out1], name="sfg")
+
+ assert sfg.__str__() == \
+ "id: add1, name: ADD1, input: [s3, s1], output: [s2]\n" + \
+ "id: c1, name: CONST, value: 3, input: [], output: [s3]\n" + \
+ "id: in1, name: INP1, input: [], output: [s1]\n" + \
+ "id: out1, name: OUT1, input: [s2], output: []\n"
+
+ def test_simple_filter(self, simple_filter):
+ assert simple_filter.__str__() == \
+ 'id: add1, name: , input: [s1, s3], output: [s4]\n' + \
+ 'id: in1, name: , input: [], output: [s1]\n' + \
+ 'id: cmul1, name: , input: [s5], output: [s3]\n' + \
+ 'id: reg1, name: , input: [s4], output: [s5, s2]\n' + \
+ 'id: out1, name: , input: [s2], output: []\n'
class TestDeepCopy:
def test_deep_copy_no_duplicates(self):
@@ -125,6 +162,19 @@ class TestDeepCopy:
assert mac_sfg_new.input(0).signals[0].source.operation is a
assert mac_sfg_new.input(1).signals[0].source.operation is b
+class TestEvaluateOutput:
+ def test_evaluate_output(self, operation_tree):
+ sfg = SFG(outputs = [Output(operation_tree)])
+ assert sfg.evaluate_output(0, []) == 5
+
+ def test_evaluate_output_large(self, large_operation_tree):
+ sfg = SFG(outputs = [Output(large_operation_tree)])
+ assert sfg.evaluate_output(0, []) == 14
+
+ def test_evaluate_output_cycle(self, operation_graph_with_cycle):
+ sfg = SFG(outputs = [Output(operation_graph_with_cycle)])
+ with pytest.raises(Exception):
+ sfg.evaluate_output(0, [])
class TestComponents:
def test_advanced_components(self):
diff --git a/test/test_signal.py b/test/test_signal.py
index cad16c9ba5b73b3d597c4e80aa666677d1909888..42086d4d5fb68eb861eee72d1351931d473b480b 100644
--- a/test/test_signal.py
+++ b/test/test_signal.py
@@ -61,27 +61,28 @@ def test_signal_creation_and_disconnction_and_connection_changing():
assert s.source is out_port
assert s.destination is in_port
-def test_signal_set_bits_pos_int(signal):
- signal.bits = 10
- assert signal.bits == 10
-
-def test_signal_set_bits_zero(signal):
- signal.bits = 0
- assert signal.bits == 0
-
-def test_signal_set_bits_neg_int(signal):
- with pytest.raises(Exception):
- signal.bits = -10
-
-def test_signal_set_bits_complex(signal):
- with pytest.raises(Exception):
- signal.bits = (2+4j)
-
-def test_signal_set_bits_float(signal):
- with pytest.raises(Exception):
- signal.bits = 3.2
-
-def test_signal_set_bits_pos_then_none(signal):
- signal.bits = 10
- signal.bits = None
- assert signal.bits is None
\ No newline at end of file
+class Bits:
+ def test_pos_int(self, signal):
+ signal.bits = 10
+ assert signal.bits == 10
+
+ def test_bits_zero(self, signal):
+ signal.bits = 0
+ assert signal.bits == 0
+
+ def test_bits_neg_int(self, signal):
+ with pytest.raises(Exception):
+ signal.bits = -10
+
+ def test_bits_complex(self, signal):
+ with pytest.raises(Exception):
+ signal.bits = (2+4j)
+
+ def test_bits_float(self, signal):
+ with pytest.raises(Exception):
+ signal.bits = 3.2
+
+ def test_bits_pos_then_none(self, signal):
+ signal.bits = 10
+ signal.bits = None
+ assert signal.bits is None
\ No newline at end of file
diff --git a/test/test_simulation.py b/test/test_simulation.py
index cb0fa79dddc662bd881b7a0692cf90e8ef3233db..70d4ede54d39dc3bfed2dac87ca3703016935a92 100644
--- a/test/test_simulation.py
+++ b/test/test_simulation.py
@@ -4,8 +4,8 @@ import numpy as np
from b_asic import SFG, Output, Simulation
-class TestSimulation:
- def test_simulate_with_lambdas_as_input(self, sfg_two_inputs_two_outputs):
+class TestRunFor:
+ def test_with_lambdas_as_input(self, sfg_two_inputs_two_outputs):
simulation = Simulation(sfg_two_inputs_two_outputs, [lambda n: n + 3, lambda n: 1 + n * 2], save_results = True)
output = simulation.run_for(101)
@@ -44,7 +44,7 @@ class TestSimulation:
assert simulation.results[3]["0"] == 13
assert simulation.results[3]["1"] == 20
- def test_simulate_with_numpy_arrays_as_input(self, sfg_two_inputs_two_outputs):
+ def test_with_numpy_arrays_as_input(self, sfg_two_inputs_two_outputs):
input0 = np.array([5, 9, 25, -5, 7])
input1 = np.array([7, 3, 3, 54, 2])
simulation = Simulation(sfg_two_inputs_two_outputs, [input0, input1])
@@ -85,8 +85,8 @@ class TestSimulation:
assert simulation.results[4]["0"] == 9
assert simulation.results[4]["1"] == 11
-
- def test_simulate_with_numpy_array_overflow(self, sfg_two_inputs_two_outputs):
+
+ def test_with_numpy_array_overflow(self, sfg_two_inputs_two_outputs):
input0 = np.array([5, 9, 25, -5, 7])
input1 = np.array([7, 3, 3, 54, 2])
simulation = Simulation(sfg_two_inputs_two_outputs, [input0, input1])
@@ -94,18 +94,7 @@ class TestSimulation:
with pytest.raises(IndexError):
simulation.run_for(1)
- def test_simulate_nested(self, sfg_nested):
- input0 = np.array([5, 9])
- input1 = np.array([7, 3])
- simulation = Simulation(sfg_nested, [input0, input1])
-
- output0 = simulation.run()
- output1 = simulation.run()
-
- assert output0[0] == 11405
- assert output1[0] == 4221
-
- def test_simulate_delay(self, sfg_delay):
+ def test_delay(self, sfg_delay):
simulation = Simulation(sfg_delay, save_results = True)
simulation.set_input(0, [5, -2, 25, -6, 7, 0])
simulation.run_for(6)
@@ -116,8 +105,20 @@ class TestSimulation:
assert simulation.results[3]["0"] == 25
assert simulation.results[4]["0"] == -6
assert simulation.results[5]["0"] == 7
+
+class TestRun:
+ def test_nested(self, sfg_nested):
+ input0 = np.array([5, 9])
+ input1 = np.array([7, 3])
+ simulation = Simulation(sfg_nested, [input0, input1])
+
+ output0 = simulation.run()
+ output1 = simulation.run()
+
+ assert output0[0] == 11405
+ assert output1[0] == 4221
- def test_simulate_accumulator(self, sfg_accumulator):
+ def test_accumulator(self, sfg_accumulator):
data_in = np.array([5, -2, 25, -6, 7, 0])
reset = np.array([0, 0, 0, 1, 0, 0])
simulation = Simulation(sfg_accumulator, [data_in, reset])
@@ -134,7 +135,7 @@ class TestSimulation:
assert output4[0] == 0
assert output5[0] == 7
- def test_simulate_simple_filter(self, simple_filter):
+ def test_simple_filter(self, simple_filter):
input0 = np.array([1, 2, 3, 4, 5])
simulation = Simulation(simple_filter, [input0], save_results=True)
output0 = [simulation.run()[0] for _ in range(len(input0))]