def test_flatten(self): arr = [[1, 2, [3]], 4] self.assertEqual(flatten(arr), [1, 2, 3, 4]) arr = [[1], 2, [3], 4] self.assertEqual(flatten(arr), [1, 2, 3, 4]) arr = [[1], 2, [3], 4, [[[[[[[6]]]]]]]] self.assertEqual(flatten(arr), [1, 2, 3, 4, 6]) arr = [1, [2, [3, [4, [5, [6, [7, [8], 9], 10], 11], 12], 13], 14]] self.assertEqual(flatten(arr), [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14])
def test_flatten_nested_all_empty(self): input = [[[]],[],[]] result = flatten(input) exptcted = [] self.assertEqual(exptcted, result)
def test_flatten_nested(self): input = [[1,2,[3]],4] result = flatten(input) exptcted = [1,2,3,4] self.assertEqual(exptcted, result)
def test_flatten_error_no_number(self): input = [1,2, ["a", "b"], [3]] result = flatten(input) exptcted = [1,2, "a", "b",3] self.assertNotEqual(exptcted, result)
def test_flatten_nested_with_empty(self): input = [[1,2,[3]],4, [],[]] result = flatten(input) exptcted = [1,2,3,4] self.assertEqual(exptcted, result)
def test_flatten(self): flat_list = flatten_array.flatten([1, [2, 3, [4], 5, [6]]]) for item in flat_list: if isinstance(item, list): flag = False else: flag = True self.assertEqual(flag, True)
def test_empty_nested_lists(self): self.assertEqual(flatten([[()]]), [])
def test_strings(self): self.assertEqual(flatten(['0', ['1', '2']]), ['0', '1', '2'])
def test_all_values_are_none(self): inputs = [None, [[[None]]], None, None, [[None, None], None], None] expected = [] self.assertEqual(flatten(inputs), expected)
def test_six_level_nesting(self): inputs = [1, [2, [[3]], [4, [[5]]], 6, 7], 8] expected = [1, 2, 3, 4, 5, 6, 7, 8] self.assertEqual(flatten(inputs), expected)
def test_6_level_nesting(self): inputs = [1, [2, [[3]], [4, [[5]]], 6, 7], 8] expected = [1, 2, 3, 4, 5, 6, 7, 8] self.assertEqual(flatten(inputs), expected)
def test_int_list(self): assert flatten(self.input_array_int_list) == [1, 2, 3]
def test_empty(self): self.assertEqual(flatten_array.flatten([]), [])
def test_diff_dataset(self): self.assertEqual(flatten_array.flatten( [2, 'loki', [3, 10]]), [2, 'loki', 3, 10])
def test_no_nesting(self): self.assertEqual(flatten([0, 1, 2]), [0, 1, 2])
def test_flattens_array_with_just_integers_present(self): inputs = [1, [2, 3, 4, 5, 6, 7], 8] expected = [1, 2, 3, 4, 5, 6, 7, 8] self.assertEqual(flatten(inputs), expected)
def test_nested_list(self): assert flatten(self.input_array_nested_list) == [1, 2, 3, 4, 5, 6]
def test_all_values_in_nested_list_are_null(self): inputs = [None, [[[None]]], None, None, [[None, None], None], None] expected = [] self.assertEqual(flatten(inputs), expected)
def test_nested_list3(self): assert flatten(self.input_array_nested_list2) == [ 1, 2, 3, 4, 5, 6, 7, 8]
def test_five_level_nesting(self): inputs = [0, 2, [[2, 3], 8, 100, 4, [[[50]]]], -2] expected = [0, 2, 2, 3, 8, 100, 4, 50, -2] self.assertEqual(flatten(inputs), expected)
def test_nested_list5(self): assert flatten(self.input_array_nested_list4) == [ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
def test_with_none_values(self): inputs = [0, 2, [[2, 3], 8, [[100]], None, [[None]]], -2] expected = [0, 2, 2, 3, 8, 100, -2] self.assertEqual(flatten(inputs), expected)
def test_empty_list(self): assert flatten([]) == []
def test_TypeError(self): with self.assertRaises(TypeError): flatten([2, 'b'])
def test_one_level_nesting(self): self.assertEqual(flatten([0, [1], 2]), [0, 1, 2])
def test_two_level_nesting(self): self.assertEqual(flatten([0, [1, [2, 3]], [4]]), [0, 1, 2, 3, 4])
def test_5_level_nesting(self): inputs = [0, 2, [[2, 3], 8, 100, 4, [[[50]]]], -2] expected = [0, 2, 2, 3, 8, 100, 4, 50, -2] assert flatten(inputs) == expected
def test_6_level_nesting(self): inputs = [1, [2, [[3]], [4, [[5]]], 6, 7], 8] expected = [1, 2, 3, 4, 5, 6, 7, 8] assert flatten(inputs) == expected
def test_5_level_nesting(self): inputs = [0, 2, [[2, 3], 8, 100, 4, [[[50]]]], -2] expected = [0, 2, 2, 3, 8, 100, 4, 50, -2] self.assertEqual(flatten(inputs), expected)
def test_all_values_in_nested_list_are_null(self): inputs = [None, [[[None]]], None, None, [[None, None], None], None] expected = [] assert flatten(inputs) == expected
def test_6_level_nest_list_with_null_values(self): inputs = [0, 2, [[2, 3], 8, [[100]], None, [[None]]], -2] expected = [0, 2, 2, 3, 8, 100, -2] self.assertEqual(flatten(inputs), expected)
def test_no_nesting(self): inputs = [0, 1, 2] expected = [0, 1, 2] assert flatten(inputs) == expected
def test_no_nesting(self): inputs = [0, 1, 2] expected = [0, 1, 2] self.assertEqual(flatten(inputs), expected)
def test_flatten_integers(self): inputs = [1, [2, 3, 4, 5, 6, 7], 8] expected = [1, 2, 3, 4, 5, 6, 7, 8] self.assertEqual(flatten(inputs), expected)