21 lines
485 B
Python
21 lines
485 B
Python
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()
|