nvim/testdir/test_flatten.vim

1 " Test for flatting list.
3   call assert_fails('call flatten(1)', 'E686:')
4   call assert_fails('call flatten({})', 'E686:')
5   call assert_fails('call flatten("string")', 'E686:')
6   call assert_fails('call flatten([], [])', 'E745:')
7   call assert_fails('call flatten([], -1)', 'E900: maxdepth')
9   call assert_equal([], flatten([]))
10   call assert_equal([], flatten([[]]))
11   call assert_equal([], flatten([[[]]]))
13   call assert_equal([1, 2, 3], flatten([1, 2, 3]))
14   call assert_equal([1, 2, 3], flatten([[1], 2, 3]))
15   call assert_equal([1, 2, 3], flatten([1, [2], 3]))
16   call assert_equal([1, 2, 3], flatten([1, 2, [3]]))
17   call assert_equal([1, 2, 3], flatten([[1], [2], 3]))
18   call assert_equal([1, 2, 3], flatten([1, [2], [3]]))
19   call assert_equal([1, 2, 3], flatten([[1], 2, [3]]))
20   call assert_equal([1, 2, 3], flatten([[1], [2], [3]]))
22   call assert_equal([1, 2, 3], flatten([[1, 2, 3], []]))
23   call assert_equal([1, 2, 3], flatten([[], [1, 2, 3]]))
24   call assert_equal([1, 2, 3], flatten([[1, 2], [], [3]]))
25   call assert_equal([1, 2, 3], flatten([[], [1, 2, 3], []]))
26   call assert_equal([1, 2, 3, 4], flatten(range(1, 4)))
29   call assert_equal([1, 2, 3, 4, 5], flatten([1, [2, [3, 4]], 5]))
30   call assert_equal([1, 2, [3, 4], 5], flatten([1, [2, [3, 4]], 5], 1))
32   call assert_equal([0, [1], 2, [3], 4], flatten([[0, [1]], 2, [[3], 4]], 1))
33   call assert_equal([1, 2, 3], flatten([[[[1]]], [2], [3]], 3))
34   call assert_equal([[1], [2], [3]], flatten([[[1], [2], [3]]], 1))
35   call assert_equal([[1]], flatten([[1]], 0))
37   " Make it flatten if the given maxdepth is larger than actual depth.
38   call assert_equal([1, 2, 3], flatten([[1, 2, 3]], 1))
39   call assert_equal([1, 2, 3], flatten([[1, 2, 3]], 2))
41   let l:list = [[1], [2], [3]]
42   call assert_equal([1, 2, 3], flatten(l:list))
43   call assert_equal([1, 2, 3], l:list)
47   call assert_equal([1, 2, l:x, 3], flatten([1, [2, l:x], 3]))
51   let l:list = [[1], [2], [3]]
52   call assert_equal([1, 2, 3], flatten(l:list))
54   call assert_equal([1, 2, 3], l:list)
56   " Tests for checking circular reference list can be flatten.
59   let l:z = flatten(l:y)
69   let l:z = flatten(l:y, 1)
79   call assert_equal([1, 2, 1, 2], flatten(l:x, 2))