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()