Removed print and added tests for slack

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):

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