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_empty_nested_lists(self):
self.assertEqual(flatten([[()]]), [])
def test_TypeError(self):
with self.assertRaises(TypeError):
flatten([2, 'b'])
def test_no_nesting(self):
self.assertEqual(flatten([0, 1, 2]), [0, 1, 2])
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_two_level_nesting(self):
self.assertEqual(flatten([0, [1, [2, 3]], [4]]), [0, 1, 2, 3, 4])
def test_strings(self):
self.assertEqual(flatten(['0', ['1', '2']]), ['0', '1', '2'])
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_one_level_nesting(self):
self.assertEqual(flatten([0, [1], 2]), [0, 1, 2])
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)
