kosmorro/test/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()