added ex2

2025-01-28 14:32:19 +01:00
parent 70daa208d0
commit 9c6bf54f20
3 changed files with 114 additions and 7 deletions
--- a/tests/test_ex2.py
+++ b/tests/test_ex2.py
@ -1,8 +1,20 @@
 import unittest

+from Ex2 import findSafestPath
+
+
 class MyTestCase(unittest.TestCase):
    def test_simple1(self):
-        pass
+        self.assertEqual(
+            findSafestPath(0, 2, [(0, 1, -1), (1, 2, 0)]),
+            [0, 0, 1, 2]
+        )
+
+    def test_simple2(self):
+        self.assertEqual(
+            findSafestPath(0, 5, [(0,1,0), (0,2,1), (2,1,-1), (1,3,-1), (2,4,-1), (3,5,1), (3,4,0), (4,5,-1)]),
+            [0, 1, 3, 5]
+        )

 if __name__ == '__main__':
    unittest.main()