lib/tests/core.py

import unittest

import kosmorrolib.core as core


class CoreTestCase(unittest.TestCase):
    def test_flatten_list(self):
        self.assertEqual(
            [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
            core.flatten_list([0, 1, 2, [3, 4, [5, 6], 7], 8, [9]]),
        )


if __name__ == "__main__":
    unittest.main()