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