tests/test_connected_components.py


1
2
3
4
5
6
7
8
9
10
11
12

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