Mercurial > vim

annotate src/testdir/test_flatten.vim @ 20766:821925509d8c v8.2.0935

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
patch 8.2.0935: flattening a list with existing code is slow Commit: https://github.com/vim/vim/commit/077a1e670ad69ef4cefc22103ca6635bd269e764 Author: Bram Moolenaar <Bram@vim.org> Date: Mon Jun 8 20:50:43 2020 +0200 patch 8.2.0935: flattening a list with existing code is slow Problem: Flattening a list with existing code is slow. Solution: Add flatten(). (Mopp, closes https://github.com/vim/vim/issues/3676)
author Bram Moolenaar <Bram@vim.org>
date Mon, 08 Jun 2020 21:00:04 +0200
parents
children 08940efa6b4e
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
20766
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
1 " Test for flatting list.
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
2 func Test_flatten()
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
3 call assert_fails('call flatten(1)', 'E686:')
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
4 call assert_fails('call flatten({})', 'E686:')
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
5 call assert_fails('call flatten("string")', 'E686:')
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
6 call assert_fails('call flatten([], [])', 'E745:')
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
7 call assert_fails('call flatten([], -1)', 'E900: maxdepth')
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
8
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
9 call assert_equal([], flatten([]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
10 call assert_equal([], flatten([[]]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
11 call assert_equal([], flatten([[[]]]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
12
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
13 call assert_equal([1, 2, 3], flatten([1, 2, 3]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
14 call assert_equal([1, 2, 3], flatten([[1], 2, 3]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
15 call assert_equal([1, 2, 3], flatten([1, [2], 3]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
16 call assert_equal([1, 2, 3], flatten([1, 2, [3]]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
17 call assert_equal([1, 2, 3], flatten([[1], [2], 3]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
18 call assert_equal([1, 2, 3], flatten([1, [2], [3]]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
19 call assert_equal([1, 2, 3], flatten([[1], 2, [3]]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
20 call assert_equal([1, 2, 3], flatten([[1], [2], [3]]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
21
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
22 call assert_equal([1, 2, 3], flatten([[1, 2, 3], []]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
23 call assert_equal([1, 2, 3], flatten([[], [1, 2, 3]]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
24 call assert_equal([1, 2, 3], flatten([[1, 2], [], [3]]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
25 call assert_equal([1, 2, 3], flatten([[], [1, 2, 3], []]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
26 call assert_equal([1, 2, 3, 4], flatten(range(1, 4)))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
27
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
28 " example in the help
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
29 call assert_equal([1, 2, 3, 4, 5], flatten([1, [2, [3, 4]], 5]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
30 call assert_equal([1, 2, [3, 4], 5], flatten([1, [2, [3, 4]], 5], 1))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
31
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
32 call assert_equal([0, [1], 2, [3], 4], flatten([[0, [1]], 2, [[3], 4]], 1))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
33 call assert_equal([1, 2, 3], flatten([[[[1]]], [2], [3]], 3))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
34 call assert_equal([[1], [2], [3]], flatten([[[1], [2], [3]]], 1))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
35 call assert_equal([[1]], flatten([[1]], 0))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
36
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
37 " Make it flatten if the given maxdepth is larger than actual depth.
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
38 call assert_equal([1, 2, 3], flatten([[1, 2, 3]], 1))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
39 call assert_equal([1, 2, 3], flatten([[1, 2, 3]], 2))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
40
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
41 let l:list = [[1], [2], [3]]
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
42 call assert_equal([1, 2, 3], flatten(l:list))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
43 call assert_equal([1, 2, 3], l:list)
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
44
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
45 " Tests for checking reference counter works well.
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
46 let l:x = {'foo': 'bar'}
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
47 call assert_equal([1, 2, l:x, 3], flatten([1, [2, l:x], 3]))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
48 call test_garbagecollect_now()
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
49 call assert_equal('bar', l:x.foo)
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
50
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
51 let l:list = [[1], [2], [3]]
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
52 call assert_equal([1, 2, 3], flatten(l:list))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
53 call test_garbagecollect_now()
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
54 call assert_equal([1, 2, 3], l:list)
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
55
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
56 " Tests for checking circular reference list can be flatten.
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
57 let l:x = [1]
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
58 let l:y = [x]
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
59 let l:z = flatten(l:y)
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
60 call assert_equal([1], l:z)
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
61 call test_garbagecollect_now()
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
62 let l:x[0] = 2
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
63 call assert_equal([2], l:x)
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
64 call assert_equal([1], l:z) " NOTE: primitive types are copied.
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
65 call assert_equal([1], l:y)
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
66
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
67 let l:x = [2]
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
68 let l:y = [1, [l:x], 3] " [1, [[2]], 3]
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
69 let l:z = flatten(l:y, 1)
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
70 call assert_equal([1, [2], 3], l:z)
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
71 let l:x[0] = 9
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
72 call assert_equal([1, [9], 3], l:z) " Reference to l:x is kept.
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
73 call assert_equal([1, [9], 3], l:y)
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
74
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
75 let l:x = [1]
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
76 let l:y = [2]
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
77 call add(x, y) " l:x = [1, [2]]
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
78 call add(y, x) " l:y = [2, [1, [...]]]
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
79 call assert_equal([1, 2, 1, 2], flatten(l:x, 2))
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
80 call assert_equal([2, l:x], l:y)
821925509d8c patch 8.2.0935: flattening a list with existing code is slow
Bram Moolenaar <Bram@vim.org>
parents:
diff changeset
81 endfunc