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.")