def test_flatten_all_iterable_types(self):
import multi_key_sort as mks
myKeys = [1, (2, 3), [4, 5], {6, 7}]
actualOutput = [isinstance(x, int) for x in mks.flatten_all(myKeys)]
self.assertTrue(all(actualOutput), "List flattening output differs from expected output.")
def test_flatten_all_n_level(self):
import multi_key_sort as mks
myKeys = [1, 2, 3, [4, [5]], [6, [7, [8, [9]]]], 10]
expectedOutput = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
actualOutput = mks.flatten_all(myKeys)
self.assertEqual(expectedOutput, actualOutput, "List flattening output differs from expected output.")
def test_flatten_all_1_level(self):
import multi_key_sort as mks
myKeys = [1, 2, 3, [4, 5]]
expectedOutput = [1, 2, 3, 4, 5]
actualOutput = mks.flatten_all(myKeys)
self.assertEqual(expectedOutput, actualOutput, "List flattening output differs from expected output.")
