import unittest from sources.acyclicity import acyclic class MyTestCase(unittest.TestCase): def test_something(self): adj = [[1], [2], [0], [0]] self.assertEqual(1, acyclic(adj)) if __name__ == '__main__': unittest.main()