import unittest

from Ex2 import findSafestPath


class MyTestCase(unittest.TestCase):
    def test_simple1(self):
        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()