From 0149e9c197f08936e2638d81b438a978c0acbb49 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Fri, 9 Feb 2024 18:15:54 +0100
Subject: [PATCH] Fixed unit test with new version of Scheduler
---
aidge_backend_cpu/unit_tests/test_scheduler.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/aidge_backend_cpu/unit_tests/test_scheduler.py b/aidge_backend_cpu/unit_tests/test_scheduler.py
index 2f174efe..17ff64cf 100644
--- a/aidge_backend_cpu/unit_tests/test_scheduler.py
+++ b/aidge_backend_cpu/unit_tests/test_scheduler.py
@@ -60,7 +60,9 @@ class test_scheduler(unittest.TestCase):
scheduler = aidge_core.SequentialScheduler(graph_view)
scheduler.generate_scheduling()
- self.assertListEqual([i.name() for i in scheduler.get_static_scheduling()], EXPECTED_SCHEDULE)
+ self.assertEqual(len(scheduler.get_static_scheduling()), 10)
+ # Do not care about the order of execution of the producers
+ self.assertListEqual([i.name() for i in scheduler.get_static_scheduling()[-3:]], EXPECTED_SCHEDULE)
def test_parallel_scheduling(self):
@@ -87,7 +89,9 @@ class test_scheduler(unittest.TestCase):
scheduler = aidge_core.SequentialScheduler(graph_view)
scheduler.generate_scheduling()
- self.assertTrue([i.name() for i in scheduler.get_static_scheduling()] in EXPECTED_SCHEDULE)
+ self.assertEqual(len(scheduler.get_static_scheduling()), 11)
+ # Do not care about the order of execution of the producers
+ self.assertTrue([i.name() for i in scheduler.get_static_scheduling()[-4:]] in EXPECTED_SCHEDULE)
if __name__ == '__main__':
unittest.main()
--
GitLab