1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
" Test for flatting list.
func Test_flatten()
call assert_fails('call flatten(1)', 'E686:')
call assert_fails('call flatten({})', 'E686:')
call assert_fails('call flatten("string")', 'E686:')
call assert_fails('call flatten([], [])', 'E745:')
call assert_fails('call flatten([], -1)', 'E900: maxdepth')
call assert_equal([], flatten([]))
call assert_equal([], flatten([[]]))
call assert_equal([], flatten([[[]]]))
call assert_equal([1, 2, 3], flatten([1, 2, 3]))
call assert_equal([1, 2, 3], flatten([[1], 2, 3]))
call assert_equal([1, 2, 3], flatten([1, [2], 3]))
call assert_equal([1, 2, 3], flatten([1, 2, [3]]))
call assert_equal([1, 2, 3], flatten([[1], [2], 3]))
call assert_equal([1, 2, 3], flatten([1, [2], [3]]))
call assert_equal([1, 2, 3], flatten([[1], 2, [3]]))
call assert_equal([1, 2, 3], flatten([[1], [2], [3]]))
call assert_equal([1, 2, 3], flatten([[1, 2, 3], []]))
call assert_equal([1, 2, 3], flatten([[], [1, 2, 3]]))
call assert_equal([1, 2, 3], flatten([[1, 2], [], [3]]))
call assert_equal([1, 2, 3], flatten([[], [1, 2, 3], []]))
call assert_equal([1, 2, 3, 4], flatten(range(1, 4)))
" example in the help
call assert_equal([1, 2, 3, 4, 5], flatten([1, [2, [3, 4]], 5]))
call assert_equal([1, 2, [3, 4], 5], flatten([1, [2, [3, 4]], 5], 1))
call assert_equal([0, [1], 2, [3], 4], flatten([[0, [1]], 2, [[3], 4]], 1))
call assert_equal([1, 2, 3], flatten([[[[1]]], [2], [3]], 3))
call assert_equal([[1], [2], [3]], flatten([[[1], [2], [3]]], 1))
call assert_equal([[1]], flatten([[1]], 0))
" Make it flatten if the given maxdepth is larger than actual depth.
call assert_equal([1, 2, 3], flatten([[1, 2, 3]], 1))
call assert_equal([1, 2, 3], flatten([[1, 2, 3]], 2))
let l:list = [[1], [2], [3]]
call assert_equal([1, 2, 3], flatten(l:list))
call assert_equal([1, 2, 3], l:list)
" Tests for checking reference counter works well.
let l:x = {'foo': 'bar'}
call assert_equal([1, 2, l:x, 3], flatten([1, [2, l:x], 3]))
call test_garbagecollect_now()
call assert_equal('bar', l:x.foo)
let l:list = [[1], [2], [3]]
call assert_equal([1, 2, 3], flatten(l:list))
call test_garbagecollect_now()
call assert_equal([1, 2, 3], l:list)
" Tests for checking circular reference list can be flattened.
let l:x = [1]
let l:y = [x]
let l:z = flatten(l:y)
call assert_equal([1], l:z)
call test_garbagecollect_now()
let l:x[0] = 2
call assert_equal([2], l:x)
call assert_equal([1], l:z) " NOTE: primitive types are copied.
call assert_equal([1], l:y)
let l:x = [2]
let l:y = [1, [l:x], 3] " [1, [[2]], 3]
let l:z = flatten(l:y, 1)
call assert_equal([1, [2], 3], l:z)
let l:x[0] = 9
call assert_equal([1, [9], 3], l:z) " Reference to l:x is kept.
call assert_equal([1, [9], 3], l:y)
let l:x = [1]
let l:y = [2]
call add(x, y) " l:x = [1, [2]]
call add(y, x) " l:y = [2, [1, [...]]]
call assert_equal([1, 2, 1, 2], flatten(l:x, 2))
call assert_equal([2, l:x], l:y)
endfunc
func Test_flattennew()
let l = [1, [2, [3, 4]], 5]
call assert_equal([1, 2, 3, 4, 5], flattennew(l))
call assert_equal([1, [2, [3, 4]], 5], l)
call assert_equal([1, 2, [3, 4], 5], flattennew(l, 1))
call assert_equal([1, [2, [3, 4]], 5], l)
endfunc
" vim: shiftwidth=2 sts=2 expandtab
|