From e25967058b677b54f722f7cc354c7e14b3eeb143 Mon Sep 17 00:00:00 2001 From: Oscar Gustafsson <oscar.gustafsson@gmail.com> Date: Thu, 28 May 2020 17:54:11 +0200 Subject: [PATCH] Removed print and added tests for slack --- b_asic/schema.py | 1 - test/test_schema.py | 11 +++++++++++ 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/b_asic/schema.py b/b_asic/schema.py index 378e9dfc..6d0c5f92 100644 --- a/b_asic/schema.py +++ b/b_asic/schema.py @@ -165,7 +165,6 @@ class Schema: self._start_times[op.graph_id] = op_start_time self._remove_delays() - print(self._laps) def plot_schedule(self) -> None: def _draw_arrow2(start, end): diff --git a/test/test_schema.py b/test/test_schema.py index 78a713a9..425fc025 100644 --- a/test/test_schema.py +++ b/test/test_schema.py @@ -65,3 +65,14 @@ class TestInit: start_times_names[op_name] = start_time assert start_times_names == {'CMUL1': 0, 'CMUL2': 5, "ADD1": 0, "CMUL3": 7} + + def test_slack_normal_latency(self, precedence_sfg_delays): + precedence_sfg_delays.set_latency_of_type(Addition.type_name(), 1) + precedence_sfg_delays.set_latency_of_type(ConstantMultiplication.type_name(), 3) + + schema = Schema(precedence_sfg_delays, scheduling_alg="ASAP") + assert schema.forward_slack(precedence_sfg_delays.find_by_name("ADD3")[0].graph_id) == 7 + assert schema.backward_slack(precedence_sfg_delays.find_by_name("ADD3")[0].graph_id) == 0 + + assert schema.forward_slack(precedence_sfg_delays.find_by_name("A2")[0].graph_id) == 0 + assert schema.backward_slack(precedence_sfg_delays.find_by_name("A2")[0].graph_id) == 16 -- GitLab