import unittest from sources.reachability import reach class TestReachability(unittest.TestCase): def testName(self): adj = [[1, 3], [], [1], [2]] self.assertEqual(1, reach(adj, 0, 3)) if __name__ == "__main__": # import sys;sys.argv = ['', 'Test.testName'] unittest.main()