diff --git a/b_asic/schema.py b/b_asic/schema.py index 378e9dfc196a5fdcc5c11340e66012253817486e..6d0c5f92e03b18fa2e6e3d92d9de44e8fd9806a8 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 78a713a9ceda574feede72f48bacf02e6a9c4025..425fc02555ffee9569f0701ad16f0f7bde3362de 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