diff --git a/test/test_sfg_generators.py b/test/test_sfg_generators.py index 699ba682979e1f30ea50b4c66ceae259fb3d1e5e..7b4755dc304fee0b5dca2cea85d1a49f37a4ea2f 100644 --- a/test/test_sfg_generators.py +++ b/test/test_sfg_generators.py @@ -57,6 +57,13 @@ def test_direct_form_fir(): len([comp for comp in sfg.components if isinstance(comp, Delay)]) == 2 ) + sfg = direct_form_fir( + (0.3, 0.4, 0.5, 0.6, 0.3), + mult_properties={'latency': 2, 'execution_time': 1}, + add_properties={'latency': 1, 'execution_time': 1}, + ) + assert sfg.critical_path() == 6 + def test_transposed_direct_form_fir(): sfg = transposed_direct_form_fir([0.3, 0.5, 0.7]) @@ -77,3 +84,10 @@ def test_transposed_direct_form_fir(): assert ( len([comp for comp in sfg.components if isinstance(comp, Delay)]) == 2 ) + + sfg = transposed_direct_form_fir( + (0.3, 0.4, 0.5, 0.6, 0.3), + mult_properties={'latency': 2, 'execution_time': 1}, + add_properties={'latency': 1, 'execution_time': 1}, + ) + assert sfg.critical_path() == 3