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