Example #1
0
 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])
Example #2
0
    def test_flatten_nested_all_empty(self):
        
        input = [[[]],[],[]]
        result = flatten(input)
        exptcted = []

        self.assertEqual(exptcted, result)
Example #3
0
    def test_flatten_nested(self):
        
        input = [[1,2,[3]],4]
        result = flatten(input)
        exptcted = [1,2,3,4]

        self.assertEqual(exptcted, result)
Example #4
0
    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)
Example #5
0
    def test_flatten_nested_with_empty(self):
        
        input = [[1,2,[3]],4, [],[]]
        result = flatten(input)
        exptcted = [1,2,3,4]

        self.assertEqual(exptcted, result)
Example #6
0
    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)
Example #7
0
 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)
Example #10
0
 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)
Example #11
0
 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)
Example #12
0
 def test_int_list(self):
     assert flatten(self.input_array_int_list) == [1, 2, 3]
Example #13
0
 def test_empty(self):
     self.assertEqual(flatten_array.flatten([]), [])
Example #14
0
 def test_diff_dataset(self):
     self.assertEqual(flatten_array.flatten(
         [2, 'loki', [3, 10]]), [2, 'loki', 3, 10])
Example #15
0
 def test_no_nesting(self):
     self.assertEqual(flatten([0, 1, 2]), [0, 1, 2])
Example #16
0
 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)
Example #17
0
 def test_nested_list(self):
     assert flatten(self.input_array_nested_list) == [1, 2, 3, 4, 5, 6]
Example #18
0
 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)
Example #19
0
 def test_nested_list3(self):
     assert flatten(self.input_array_nested_list2) == [
         1, 2, 3, 4, 5, 6, 7, 8]
Example #20
0
 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)
Example #21
0
 def test_nested_list5(self):
     assert flatten(self.input_array_nested_list4) == [
         1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
Example #22
0
 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)
Example #23
0
 def test_empty_list(self):
     assert flatten([]) == []
Example #24
0
 def test_empty_nested_lists(self):
     self.assertEqual(flatten([[()]]), [])
Example #25
0
 def test_TypeError(self):
     with self.assertRaises(TypeError):
         flatten([2, 'b'])
Example #26
0
 def test_no_nesting(self):
     self.assertEqual(flatten([0, 1, 2]), [0, 1, 2])
Example #27
0
 def test_one_level_nesting(self):
     self.assertEqual(flatten([0, [1], 2]), [0, 1, 2])
Example #28
0
 def test_two_level_nesting(self):
     self.assertEqual(flatten([0, [1, [2, 3]], [4]]), [0, 1, 2, 3, 4])
Example #29
0
 def test_two_level_nesting(self):
     self.assertEqual(flatten([0, [1, [2, 3]], [4]]), [0, 1, 2, 3, 4])
Example #30
0
 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
Example #32
0
 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
Example #34
0
 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
Example #36
0
 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
Example #38
0
 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)