blob: a16bae5ee32821b56257a16ca9bedd8ae2ea1e63 [file] [log] [blame]
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02001" Tests for fuzzy matching
2
3source shared.vim
4source check.vim
Bram Moolenaarcaf642c2023-04-29 21:38:04 +01005source term_util.vim
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +02006
7" Test for matchfuzzy()
8func Test_matchfuzzy()
9 call assert_fails('call matchfuzzy(10, "abc")', 'E686:')
10 call assert_fails('call matchfuzzy(["abc"], [])', 'E730:')
11 call assert_fails("let x = matchfuzzy(test_null_list(), 'foo')", 'E686:')
12 call assert_fails('call matchfuzzy(["abc"], test_null_string())', 'E475:')
13 call assert_equal([], matchfuzzy([], 'abc'))
14 call assert_equal([], matchfuzzy(['abc'], ''))
15 call assert_equal(['abc'], matchfuzzy(['abc', 10], 'ac'))
16 call assert_equal([], matchfuzzy([10, 20], 'ac'))
17 call assert_equal(['abc'], matchfuzzy(['abc'], 'abc'))
18 call assert_equal(['crayon', 'camera'], matchfuzzy(['camera', 'crayon'], 'cra'))
19 call assert_equal(['aabbaa', 'aaabbbaaa', 'aaaabbbbaaaa', 'aba'], matchfuzzy(['aba', 'aabbaa', 'aaabbbaaa', 'aaaabbbbaaaa'], 'aa'))
20 call assert_equal(['one'], matchfuzzy(['one', 'two'], 'one'))
21 call assert_equal(['oneTwo', 'onetwo'], matchfuzzy(['onetwo', 'oneTwo'], 'oneTwo'))
Bram Moolenaare9f9f162020-10-20 19:01:30 +020022 call assert_equal(['onetwo', 'one_two'], matchfuzzy(['onetwo', 'one_two'], 'oneTwo'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +020023 call assert_equal(['aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'], matchfuzzy(['aaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'], 'aa'))
24 call assert_equal(256, matchfuzzy([repeat('a', 256)], repeat('a', 256))[0]->len())
25 call assert_equal([], matchfuzzy([repeat('a', 300)], repeat('a', 257)))
glepnir5a049992024-12-26 15:38:39 +010026 " full match has highest score
27 call assert_equal(['Cursor', 'lCursor'], matchfuzzy(["hello", "lCursor", "Cursor"], "Cursor"))
Bram Moolenaar8ded5b62020-10-23 16:50:30 +020028 " matches with same score should not be reordered
29 let l = ['abc1', 'abc2', 'abc3']
30 call assert_equal(l, l->matchfuzzy('abc'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +020031
32 " Tests for match preferences
33 " preference for camel case match
34 call assert_equal(['oneTwo', 'onetwo'], ['onetwo', 'oneTwo']->matchfuzzy('onetwo'))
35 " preference for match after a separator (_ or space)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +020036 call assert_equal(['onetwo', 'one_two', 'one two'], ['onetwo', 'one_two', 'one two']->matchfuzzy('onetwo'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +020037 " preference for leading letter match
38 call assert_equal(['onetwo', 'xonetwo'], ['xonetwo', 'onetwo']->matchfuzzy('onetwo'))
39 " preference for sequential match
40 call assert_equal(['onetwo', 'oanbectdweo'], ['oanbectdweo', 'onetwo']->matchfuzzy('onetwo'))
41 " non-matching leading letter(s) penalty
42 call assert_equal(['xonetwo', 'xxonetwo'], ['xxonetwo', 'xonetwo']->matchfuzzy('onetwo'))
43 " total non-matching letter(s) penalty
44 call assert_equal(['one', 'onex', 'onexx'], ['onexx', 'one', 'onex']->matchfuzzy('one'))
Bram Moolenaare9f9f162020-10-20 19:01:30 +020045 " prefer complete matches over separator matches
46 call assert_equal(['.vim/vimrc', '.vim/vimrc_colors', '.vim/v_i_m_r_c'], ['.vim/vimrc', '.vim/vimrc_colors', '.vim/v_i_m_r_c']->matchfuzzy('vimrc'))
Bram Moolenaar8ded5b62020-10-23 16:50:30 +020047 " gap penalty
48 call assert_equal(['xxayybxxxx', 'xxayyybxxx', 'xxayyyybxx'], ['xxayyyybxx', 'xxayyybxxx', 'xxayybxxxx']->matchfuzzy('ab'))
Bram Moolenaardcdd42a2020-10-29 18:58:01 +010049 " path separator vs word separator
50 call assert_equal(['color/setup.vim', 'color\\setup.vim', 'color setup.vim', 'color_setup.vim', 'colorsetup.vim'], matchfuzzy(['colorsetup.vim', 'color setup.vim', 'color/setup.vim', 'color_setup.vim', 'color\\setup.vim'], 'setup.vim'))
Bram Moolenaar8ded5b62020-10-23 16:50:30 +020051
52 " match multiple words (separated by space)
53 call assert_equal(['foo bar baz'], ['foo bar baz', 'foo', 'foo bar', 'baz bar']->matchfuzzy('baz foo'))
54 call assert_equal([], ['foo bar baz', 'foo', 'foo bar', 'baz bar']->matchfuzzy('one two'))
55 call assert_equal([], ['foo bar']->matchfuzzy(" \t "))
56
57 " test for matching a sequence of words
58 call assert_equal(['bar foo'], ['foo bar', 'bar foo', 'foobar', 'barfoo']->matchfuzzy('bar foo', {'matchseq' : 1}))
59 call assert_equal([#{text: 'two one'}], [#{text: 'one two'}, #{text: 'two one'}]->matchfuzzy('two one', #{key: 'text', matchseq: v:true}))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +020060
61 %bw!
62 eval ['somebuf', 'anotherone', 'needle', 'yetanotherone']->map({_, v -> bufadd(v) + bufload(v)})
Bram Moolenaar22e7e862022-07-02 20:48:01 +010063 let l = getbufinfo()->map({_, v -> fnamemodify(v.name, ':t')})->matchfuzzy('ndl')
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +020064 call assert_equal(1, len(l))
65 call assert_match('needle', l[0])
66
Bram Moolenaar8ded5b62020-10-23 16:50:30 +020067 " Test for fuzzy matching dicts
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +020068 let l = [{'id' : 5, 'val' : 'crayon'}, {'id' : 6, 'val' : 'camera'}]
69 call assert_equal([{'id' : 6, 'val' : 'camera'}], matchfuzzy(l, 'cam', {'text_cb' : {v -> v.val}}))
70 call assert_equal([{'id' : 6, 'val' : 'camera'}], matchfuzzy(l, 'cam', {'key' : 'val'}))
71 call assert_equal([], matchfuzzy(l, 'day', {'text_cb' : {v -> v.val}}))
72 call assert_equal([], matchfuzzy(l, 'day', {'key' : 'val'}))
Yegappan Lakshmanan04c4c572022-08-30 19:48:24 +010073 call assert_fails("let x = matchfuzzy(l, 'cam', 'random')", 'E1206:')
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +020074 call assert_equal([], matchfuzzy(l, 'day', {'text_cb' : {v -> []}}))
75 call assert_equal([], matchfuzzy(l, 'day', {'text_cb' : {v -> 1}}))
76 call assert_fails("let x = matchfuzzy(l, 'day', {'text_cb' : {a, b -> 1}})", 'E119:')
77 call assert_equal([], matchfuzzy(l, 'cam'))
78 call assert_fails("let x = matchfuzzy(l, 'cam', {'text_cb' : []})", 'E921:')
zeertzjqc4815c12025-03-18 20:28:00 +010079 call assert_fails("let x = matchfuzzy(l, 'foo', {'key' : 123})", 'E475: Invalid value for argument key: 123')
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +020080 call assert_fails("let x = matchfuzzy(l, 'foo', {'key' : []})", 'E730:')
Yegappan Lakshmanan04c4c572022-08-30 19:48:24 +010081 call assert_fails("let x = matchfuzzy(l, 'cam', test_null_dict())", 'E1297:')
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +020082 call assert_fails("let x = matchfuzzy(l, 'foo', {'key' : test_null_string()})", 'E475:')
83 call assert_fails("let x = matchfuzzy(l, 'foo', {'text_cb' : test_null_function()})", 'E475:')
Bram Moolenaar8ded5b62020-10-23 16:50:30 +020084 " matches with same score should not be reordered
85 let l = [#{text: 'abc', id: 1}, #{text: 'abc', id: 2}, #{text: 'abc', id: 3}]
86 call assert_equal(l, l->matchfuzzy('abc', #{key: 'text'}))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +020087
88 let l = [{'id' : 5, 'name' : 'foo'}, {'id' : 6, 'name' : []}, {'id' : 7}]
89 call assert_fails("let x = matchfuzzy(l, 'foo', {'key' : 'name'})", 'E730:')
90
zeertzjq85627732025-03-17 21:06:02 +010091 " camelcase
zeertzjq85627732025-03-17 21:06:02 +010092 call assert_equal(['lCursor', 'shCurlyIn', 'shCurlyError', 'TracesCursor', 'Cursor', 'CurSearch', 'CursorLine'],
93 \ matchfuzzy(['Cursor', 'lCursor', 'shCurlyIn', 'shCurlyError', 'TracesCursor', 'CurSearch', 'CursorLine'], 'Cur'))
zeertzjqc4815c12025-03-18 20:28:00 +010094 call assert_equal(['lCursor', 'shCurlyIn', 'shCurlyError', 'TracesCursor', 'Cursor', 'CurSearch', 'CursorLine'],
95 \ matchfuzzy(['Cursor', 'lCursor', 'shCurlyIn', 'shCurlyError', 'TracesCursor', 'CurSearch', 'CursorLine'], 'Cur', {"camelcase": v:true}))
96 call assert_equal(['Cursor', 'CurSearch', 'CursorLine', 'lCursor', 'shCurlyIn', 'shCurlyError', 'TracesCursor'],
97 \ matchfuzzy(['Cursor', 'lCursor', 'shCurlyIn', 'shCurlyError', 'TracesCursor', 'CurSearch', 'CursorLine'], 'Cur', {"camelcase": v:false}))
98 call assert_equal(['things', 'sThings', 'thisThings'],
99 \ matchfuzzy(['things','sThings', 'thisThings'], 'thin', {'camelcase': v:false}))
100 call assert_fails("let x = matchfuzzy([], 'foo', {'camelcase': []})", 'E475: Invalid value for argument camelcase')
glepnir28e40a72025-03-16 21:24:22 +0100101
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200102 " Test in latin1 encoding
103 let save_enc = &encoding
104 set encoding=latin1
105 call assert_equal(['abc'], matchfuzzy(['abc'], 'abc'))
106 let &encoding = save_enc
107endfunc
108
Bram Moolenaar8ded5b62020-10-23 16:50:30 +0200109" Test for the matchfuzzypos() function
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200110func Test_matchfuzzypos()
glepnir9dfc7e52025-01-21 22:33:13 +0100111 call assert_equal([['curl', 'world'], [[2,3], [2,3]], [178, 177]], matchfuzzypos(['world', 'curl'], 'rl'))
112 call assert_equal([['curl', 'world'], [[2,3], [2,3]], [178, 177]], matchfuzzypos(['world', 'one', 'curl'], 'rl'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200113 call assert_equal([['hello', 'hello world hello world'],
glepnir9dfc7e52025-01-21 22:33:13 +0100114 \ [[0, 1, 2, 3, 4], [0, 1, 2, 3, 4]], [500, 382]],
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200115 \ matchfuzzypos(['hello world hello world', 'hello', 'world'], 'hello'))
glepnir9dfc7e52025-01-21 22:33:13 +0100116 call assert_equal([['aaaaaaa'], [[0, 1, 2]], [266]], matchfuzzypos(['aaaaaaa'], 'aaa'))
117 call assert_equal([['a b'], [[0, 3]], [269]], matchfuzzypos(['a b'], 'a b'))
118 call assert_equal([['a b'], [[0, 3]], [269]], matchfuzzypos(['a b'], 'a b'))
119 call assert_equal([['a b'], [[0]], [137]], matchfuzzypos(['a b'], ' a '))
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100120 call assert_equal([[], [], []], matchfuzzypos(['a b'], ' '))
121 call assert_equal([[], [], []], matchfuzzypos(['world', 'curl'], 'ab'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200122 let x = matchfuzzypos([repeat('a', 256)], repeat('a', 256))
123 call assert_equal(range(256), x[1][0])
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100124 call assert_equal([[], [], []], matchfuzzypos([repeat('a', 300)], repeat('a', 257)))
125 call assert_equal([[], [], []], matchfuzzypos([], 'abc'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200126
127 " match in a long string
Girish Palya328332b2025-06-09 20:43:03 +0200128 call assert_equal([[repeat('x', 300) .. 'abc'], [[300, 301, 302]], [155]],
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200129 \ matchfuzzypos([repeat('x', 300) .. 'abc'], 'abc'))
130
131 " preference for camel case match
glepnir9dfc7e52025-01-21 22:33:13 +0100132 call assert_equal([['xabcxxaBc'], [[6, 7, 8]], [269]], matchfuzzypos(['xabcxxaBc'], 'abc'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200133 " preference for match after a separator (_ or space)
glepnir9dfc7e52025-01-21 22:33:13 +0100134 call assert_equal([['xabx_ab'], [[5, 6]], [195]], matchfuzzypos(['xabx_ab'], 'ab'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200135 " preference for leading letter match
glepnir9dfc7e52025-01-21 22:33:13 +0100136 call assert_equal([['abcxabc'], [[0, 1]], [200]], matchfuzzypos(['abcxabc'], 'ab'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200137 " preference for sequential match
glepnir9dfc7e52025-01-21 22:33:13 +0100138 call assert_equal([['aobncedone'], [[7, 8, 9]], [233]], matchfuzzypos(['aobncedone'], 'one'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200139 " best recursive match
glepnir9dfc7e52025-01-21 22:33:13 +0100140 call assert_equal([['xoone'], [[2, 3, 4]], [243]], matchfuzzypos(['xoone'], 'one'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200141
Bram Moolenaar8ded5b62020-10-23 16:50:30 +0200142 " match multiple words (separated by space)
glepnir9dfc7e52025-01-21 22:33:13 +0100143 call assert_equal([['foo bar baz'], [[8, 9, 10, 0, 1, 2]], [519]], ['foo bar baz', 'foo', 'foo bar', 'baz bar']->matchfuzzypos('baz foo'))
zeertzjq9af2bc02022-05-11 14:15:37 +0100144 call assert_equal([[], [], []], ['foo bar baz', 'foo', 'foo bar', 'baz bar']->matchfuzzypos('baz foo', {'matchseq': 1}))
glepnir9dfc7e52025-01-21 22:33:13 +0100145 call assert_equal([['foo bar baz'], [[0, 1, 2, 8, 9, 10]], [519]], ['foo bar baz', 'foo', 'foo bar', 'baz bar']->matchfuzzypos('foo baz'))
146 call assert_equal([['foo bar baz'], [[0, 1, 2, 3, 4, 5, 10]], [476]], ['foo bar baz', 'foo', 'foo bar', 'baz bar']->matchfuzzypos('foo baz', {'matchseq': 1}))
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100147 call assert_equal([[], [], []], ['foo bar baz', 'foo', 'foo bar', 'baz bar']->matchfuzzypos('one two'))
148 call assert_equal([[], [], []], ['foo bar']->matchfuzzypos(" \t "))
glepnir9dfc7e52025-01-21 22:33:13 +0100149 call assert_equal([['grace'], [[1, 2, 3, 4, 2, 3, 4, 0, 1, 2, 3, 4]], [1057]], ['grace']->matchfuzzypos('race ace grace'))
Bram Moolenaar8ded5b62020-10-23 16:50:30 +0200150
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200151 let l = [{'id' : 5, 'val' : 'crayon'}, {'id' : 6, 'val' : 'camera'}]
glepnir9dfc7e52025-01-21 22:33:13 +0100152 call assert_equal([[{'id' : 6, 'val' : 'camera'}], [[0, 1, 2]], [267]],
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200153 \ matchfuzzypos(l, 'cam', {'text_cb' : {v -> v.val}}))
glepnir9dfc7e52025-01-21 22:33:13 +0100154 call assert_equal([[{'id' : 6, 'val' : 'camera'}], [[0, 1, 2]], [267]],
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200155 \ matchfuzzypos(l, 'cam', {'key' : 'val'}))
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100156 call assert_equal([[], [], []], matchfuzzypos(l, 'day', {'text_cb' : {v -> v.val}}))
157 call assert_equal([[], [], []], matchfuzzypos(l, 'day', {'key' : 'val'}))
Yegappan Lakshmanan04c4c572022-08-30 19:48:24 +0100158 call assert_fails("let x = matchfuzzypos(l, 'cam', 'random')", 'E1206:')
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100159 call assert_equal([[], [], []], matchfuzzypos(l, 'day', {'text_cb' : {v -> []}}))
160 call assert_equal([[], [], []], matchfuzzypos(l, 'day', {'text_cb' : {v -> 1}}))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200161 call assert_fails("let x = matchfuzzypos(l, 'day', {'text_cb' : {a, b -> 1}})", 'E119:')
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100162 call assert_equal([[], [], []], matchfuzzypos(l, 'cam'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200163 call assert_fails("let x = matchfuzzypos(l, 'cam', {'text_cb' : []})", 'E921:')
zeertzjqc4815c12025-03-18 20:28:00 +0100164 call assert_fails("let x = matchfuzzypos(l, 'foo', {'key' : 123})", 'E475: Invalid value for argument key: 123')
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200165 call assert_fails("let x = matchfuzzypos(l, 'foo', {'key' : []})", 'E730:')
Yegappan Lakshmanan04c4c572022-08-30 19:48:24 +0100166 call assert_fails("let x = matchfuzzypos(l, 'cam', test_null_dict())", 'E1297:')
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200167 call assert_fails("let x = matchfuzzypos(l, 'foo', {'key' : test_null_string()})", 'E475:')
168 call assert_fails("let x = matchfuzzypos(l, 'foo', {'text_cb' : test_null_function()})", 'E475:')
169
glepnir9dfc7e52025-01-21 22:33:13 +0100170 " case match
171 call assert_equal([['Match', 'match'], [[0, 1], [0, 1]], [202, 177]], matchfuzzypos(['match', 'Match'], 'Ma'))
172 call assert_equal([['match', 'Match'], [[0, 1], [0, 1]], [202, 177]], matchfuzzypos(['Match', 'match'], 'ma'))
173 " CamelCase has high weight even case match
174 call assert_equal(['MyTestCase', 'mytestcase'], matchfuzzy(['mytestcase', 'MyTestCase'], 'mtc'))
175 call assert_equal(['MyTestCase', 'mytestcase'], matchfuzzy(['MyTestCase', 'mytestcase'], 'mtc'))
176 call assert_equal(['MyTest', 'Mytest', 'mytest', ],matchfuzzy(['Mytest', 'mytest', 'MyTest'], 'MyT'))
177 call assert_equal(['CamelCaseMatchIngAlg', 'camelCaseMatchingAlg', 'camelcasematchingalg'],
178 \ matchfuzzy(['CamelCaseMatchIngAlg', 'camelcasematchingalg', 'camelCaseMatchingAlg'], 'CamelCase'))
179 call assert_equal(['CamelCaseMatchIngAlg', 'camelCaseMatchingAlg', 'camelcasematchingalg'],
180 \ matchfuzzy(['CamelCaseMatchIngAlg', 'camelcasematchingalg', 'camelCaseMatchingAlg'], 'CamelcaseM'))
181
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200182 let l = [{'id' : 5, 'name' : 'foo'}, {'id' : 6, 'name' : []}, {'id' : 7}]
183 call assert_fails("let x = matchfuzzypos(l, 'foo', {'key' : 'name'})", 'E730:')
glepnir28e40a72025-03-16 21:24:22 +0100184
zeertzjq85627732025-03-17 21:06:02 +0100185 " camelcase
glepnir28e40a72025-03-16 21:24:22 +0100186 call assert_equal([['lCursor', 'shCurlyIn', 'shCurlyError', 'TracesCursor', 'Cursor', 'CurSearch', 'CursorLine'], [[1, 2, 3], [2, 3, 4], [2, 3, 4], [6, 7, 8], [0, 1, 2], [0, 1, 2], [0, 1, 2]], [318, 311, 308, 303, 267, 264, 263]],
zeertzjq85627732025-03-17 21:06:02 +0100187 \ matchfuzzypos(['Cursor', 'lCursor', 'shCurlyIn', 'shCurlyError', 'TracesCursor', 'CurSearch', 'CursorLine'], 'Cur'))
zeertzjqc4815c12025-03-18 20:28:00 +0100188 call assert_equal([['lCursor', 'shCurlyIn', 'shCurlyError', 'TracesCursor', 'Cursor', 'CurSearch', 'CursorLine'], [[1, 2, 3], [2, 3, 4], [2, 3, 4], [6, 7, 8], [0, 1, 2], [0, 1, 2], [0, 1, 2]], [318, 311, 308, 303, 267, 264, 263]],
189 \ matchfuzzypos(['Cursor', 'lCursor', 'shCurlyIn', 'shCurlyError', 'TracesCursor', 'CurSearch', 'CursorLine'], 'Cur', {"camelcase": v:true}))
glepnir28e40a72025-03-16 21:24:22 +0100190 call assert_equal([['Cursor', 'CurSearch', 'CursorLine', 'lCursor', 'shCurlyIn', 'shCurlyError', 'TracesCursor'], [[0, 1, 2], [0, 1, 2], [0, 1, 2], [1, 2, 3], [2, 3, 4], [2, 3, 4], [6, 7, 8]], [267, 264, 263, 246, 239, 236, 231]],
191 \ matchfuzzypos(['Cursor', 'lCursor', 'shCurlyIn', 'shCurlyError', 'TracesCursor', 'CurSearch', 'CursorLine'], 'Cur', {"camelcase": v:false}))
192 call assert_equal([['things', 'sThings', 'thisThings'], [[0, 1, 2, 3], [1, 2, 3, 4], [0, 1, 2, 7]], [333, 287, 279]],
193 \ matchfuzzypos(['things','sThings', 'thisThings'], 'thin', {'camelcase': v:false}))
zeertzjqc4815c12025-03-18 20:28:00 +0100194 call assert_fails("let x = matchfuzzypos([], 'foo', {'camelcase': []})", 'E475: Invalid value for argument camelcase')
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200195endfunc
196
Bram Moolenaar8ded5b62020-10-23 16:50:30 +0200197" Test for matchfuzzy() with multibyte characters
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200198func Test_matchfuzzy_mbyte()
199 CheckFeature multi_lang
200 call assert_equal(['ンヹㄇヺヴ'], matchfuzzy(['ンヹㄇヺヴ'], 'ヹヺ'))
201 " reverse the order of characters
202 call assert_equal([], matchfuzzy(['ンヹㄇヺヴ'], 'ヺヹ'))
203 call assert_equal(['αβΩxxx', 'xαxβxΩx'],
204 \ matchfuzzy(['αβΩxxx', 'xαxβxΩx'], 'αβΩ'))
205 call assert_equal(['ππbbππ', 'πππbbbπππ', 'ππππbbbbππππ', 'πbπ'],
206 \ matchfuzzy(['πbπ', 'ππbbππ', 'πππbbbπππ', 'ππππbbbbππππ'], 'ππ'))
207
Bram Moolenaar8ded5b62020-10-23 16:50:30 +0200208 " match multiple words (separated by space)
209 call assert_equal(['세 마리의 작은 돼지'], ['세 마리의 작은 돼지', '마리의', '마리의 작은', '작은 돼지']->matchfuzzy('돼지 마리의'))
210 call assert_equal([], ['세 마리의 작은 돼지', '마리의', '마리의 작은', '작은 돼지']->matchfuzzy('파란 하늘'))
211
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200212 " preference for camel case match
213 call assert_equal(['oneĄwo', 'oneąwo'],
214 \ ['oneąwo', 'oneĄwo']->matchfuzzy('oneąwo'))
Bram Moolenaare9f9f162020-10-20 19:01:30 +0200215 " preference for complete match then match after separator (_ or space)
Bram Moolenaar8ded5b62020-10-23 16:50:30 +0200216 call assert_equal(['ⅠⅡabㄟㄠ'] + sort(['ⅠⅡa_bㄟㄠ', 'ⅠⅡa bㄟㄠ']),
Bram Moolenaare9f9f162020-10-20 19:01:30 +0200217 \ ['ⅠⅡabㄟㄠ', 'ⅠⅡa bㄟㄠ', 'ⅠⅡa_bㄟㄠ']->matchfuzzy('ⅠⅡabㄟㄠ'))
Bram Moolenaar8ded5b62020-10-23 16:50:30 +0200218 " preference for match after a separator (_ or space)
219 call assert_equal(['ㄓㄔabㄟㄠ', 'ㄓㄔa_bㄟㄠ', 'ㄓㄔa bㄟㄠ'],
220 \ ['ㄓㄔa_bㄟㄠ', 'ㄓㄔa bㄟㄠ', 'ㄓㄔabㄟㄠ']->matchfuzzy('ㄓㄔabㄟㄠ'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200221 " preference for leading letter match
222 call assert_equal(['ŗŝţũŵż', 'xŗŝţũŵż'],
223 \ ['xŗŝţũŵż', 'ŗŝţũŵż']->matchfuzzy('ŗŝţũŵż'))
224 " preference for sequential match
225 call assert_equal(['ㄞㄡㄤfffifl', 'ㄞaㄡbㄤcffdfiefl'],
226 \ ['ㄞaㄡbㄤcffdfiefl', 'ㄞㄡㄤfffifl']->matchfuzzy('ㄞㄡㄤfffifl'))
227 " non-matching leading letter(s) penalty
228 call assert_equal(['xㄞㄡㄤfffifl', 'xxㄞㄡㄤfffifl'],
229 \ ['xxㄞㄡㄤfffifl', 'xㄞㄡㄤfffifl']->matchfuzzy('ㄞㄡㄤfffifl'))
230 " total non-matching letter(s) penalty
231 call assert_equal(['ŗŝţ', 'ŗŝţx', 'ŗŝţxx'],
232 \ ['ŗŝţxx', 'ŗŝţ', 'ŗŝţx']->matchfuzzy('ŗŝţ'))
233endfunc
234
Bram Moolenaar8ded5b62020-10-23 16:50:30 +0200235" Test for matchfuzzypos() with multibyte characters
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200236func Test_matchfuzzypos_mbyte()
237 CheckFeature multi_lang
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100238 call assert_equal([['こんにちは世界'], [[0, 1, 2, 3, 4]], [273]],
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200239 \ matchfuzzypos(['こんにちは世界'], 'こんにちは'))
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100240 call assert_equal([['ンヹㄇヺヴ'], [[1, 3]], [88]], matchfuzzypos(['ンヹㄇヺヴ'], 'ヹヺ'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200241 " reverse the order of characters
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100242 call assert_equal([[], [], []], matchfuzzypos(['ンヹㄇヺヴ'], 'ヺヹ'))
glepnir9dfc7e52025-01-21 22:33:13 +0100243 call assert_equal([['αβΩxxx', 'xαxβxΩx'], [[0, 1, 2], [1, 3, 5]], [252, 143]],
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200244 \ matchfuzzypos(['αβΩxxx', 'xαxβxΩx'], 'αβΩ'))
245 call assert_equal([['ππbbππ', 'πππbbbπππ', 'ππππbbbbππππ', 'πbπ'],
glepnir9dfc7e52025-01-21 22:33:13 +0100246 \ [[0, 1], [0, 1], [0, 1], [0, 2]], [176, 173, 170, 135]],
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200247 \ matchfuzzypos(['πbπ', 'ππbbππ', 'πππbbbπππ', 'ππππbbbbππππ'], 'ππ'))
glepnir9dfc7e52025-01-21 22:33:13 +0100248 call assert_equal([['ααααααα'], [[0, 1, 2]], [216]],
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200249 \ matchfuzzypos(['ααααααα'], 'ααα'))
250
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100251 call assert_equal([[], [], []], matchfuzzypos(['ンヹㄇ', 'ŗŝţ'], 'fffifl'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200252 let x = matchfuzzypos([repeat('Ψ', 256)], repeat('Ψ', 256))
253 call assert_equal(range(256), x[1][0])
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100254 call assert_equal([[], [], []], matchfuzzypos([repeat('✓', 300)], repeat('✓', 257)))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200255
Bram Moolenaar8ded5b62020-10-23 16:50:30 +0200256 " match multiple words (separated by space)
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100257 call assert_equal([['세 마리의 작은 돼지'], [[9, 10, 2, 3, 4]], [328]], ['세 마리의 작은 돼지', '마리의', '마리의 작은', '작은 돼지']->matchfuzzypos('돼지 마리의'))
258 call assert_equal([[], [], []], ['세 마리의 작은 돼지', '마리의', '마리의 작은', '작은 돼지']->matchfuzzypos('파란 하늘'))
Bram Moolenaar8ded5b62020-10-23 16:50:30 +0200259
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200260 " match in a long string
Girish Palya328332b2025-06-09 20:43:03 +0200261 call assert_equal([[repeat('ぶ', 300) .. 'ẼẼẼ'], [[300, 301, 302]], [105]],
Bram Moolenaar8ded5b62020-10-23 16:50:30 +0200262 \ matchfuzzypos([repeat('ぶ', 300) .. 'ẼẼẼ'], 'ẼẼẼ'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200263 " preference for camel case match
glepnir9dfc7e52025-01-21 22:33:13 +0100264 call assert_equal([['xѳѵҁxxѳѴҁ'], [[6, 7, 8]], [219]], matchfuzzypos(['xѳѵҁxxѳѴҁ'], 'ѳѵҁ'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200265 " preference for match after a separator (_ or space)
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100266 call assert_equal([['xちだx_ちだ'], [[5, 6]], [145]], matchfuzzypos(['xちだx_ちだ'], 'ちだ'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200267 " preference for leading letter match
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100268 call assert_equal([['ѳѵҁxѳѵҁ'], [[0, 1]], [150]], matchfuzzypos(['ѳѵҁxѳѵҁ'], 'ѳѵ'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200269 " preference for sequential match
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100270 call assert_equal([['aンbヹcㄇdンヹㄇ'], [[7, 8, 9]], [158]], matchfuzzypos(['aンbヹcㄇdンヹㄇ'], 'ンヹㄇ'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200271 " best recursive match
Bram Moolenaar9d19e4f2021-01-02 18:31:32 +0100272 call assert_equal([['xффйд'], [[2, 3, 4]], [168]], matchfuzzypos(['xффйд'], 'фйд'))
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200273endfunc
274
Yasuhiro Matsumoto9029a6e2022-04-16 12:35:35 +0100275" Test for matchfuzzy() with limit
276func Test_matchfuzzy_limit()
277 let x = ['1', '2', '3', '2']
278 call assert_equal(['2', '2'], x->matchfuzzy('2'))
279 call assert_equal(['2', '2'], x->matchfuzzy('2', #{}))
280 call assert_equal(['2', '2'], x->matchfuzzy('2', #{limit: 0}))
281 call assert_equal(['2'], x->matchfuzzy('2', #{limit: 1}))
282 call assert_equal(['2', '2'], x->matchfuzzy('2', #{limit: 2}))
283 call assert_equal(['2', '2'], x->matchfuzzy('2', #{limit: 3}))
284 call assert_fails("call matchfuzzy(x, '2', #{limit: '2'})", 'E475:')
zeertzjqc4815c12025-03-18 20:28:00 +0100285 call assert_fails("call matchfuzzy(x, '2', #{limit: []})", 'E475:')
Kazuyuki Miyagi47f1a552022-06-17 18:30:03 +0100286
287 let l = [{'id': 5, 'val': 'crayon'}, {'id': 6, 'val': 'camera'}]
288 call assert_equal([{'id': 5, 'val': 'crayon'}, {'id': 6, 'val': 'camera'}], l->matchfuzzy('c', #{text_cb: {v -> v.val}}))
289 call assert_equal([{'id': 5, 'val': 'crayon'}, {'id': 6, 'val': 'camera'}], l->matchfuzzy('c', #{key: 'val'}))
290 call assert_equal([{'id': 5, 'val': 'crayon'}, {'id': 6, 'val': 'camera'}], l->matchfuzzy('c', #{text_cb: {v -> v.val}, limit: 0}))
291 call assert_equal([{'id': 5, 'val': 'crayon'}, {'id': 6, 'val': 'camera'}], l->matchfuzzy('c', #{key: 'val', limit: 0}))
292 call assert_equal([{'id': 5, 'val': 'crayon'}], l->matchfuzzy('c', #{text_cb: {v -> v.val}, limit: 1}))
293 call assert_equal([{'id': 5, 'val': 'crayon'}], l->matchfuzzy('c', #{key: 'val', limit: 1}))
Yasuhiro Matsumoto9029a6e2022-04-16 12:35:35 +0100294endfunc
295
Bram Moolenaarcaf642c2023-04-29 21:38:04 +0100296" This was using uninitialized memory
297func Test_matchfuzzy_initialized()
298 CheckRunVimInTerminal
299
300 " This can take a very long time (esp. when using valgrind). Run in a
301 " separate Vim instance and kill it after two seconds. We only check for
302 " memory errors.
303 let lines =<< trim END
304 lvimgrep [ss [fg*
305 END
306 call writefile(lines, 'XTest_matchfuzzy', 'D')
307
308 let buf = RunVimInTerminal('-u NONE -X -Z', {})
309 call term_sendkeys(buf, ":source XTest_matchfuzzy\n")
310 call TermWait(buf, 2000)
311
312 let job = term_getjob(buf)
313 if job_status(job) == "run"
314 call job_stop(job, "int")
315 call TermWait(buf, 50)
316 endif
317
318 " clean up
319 call StopVimInTerminal(buf)
320endfunc
321
Bram Moolenaar4f73b8e2020-09-22 20:33:50 +0200322" vim: shiftwidth=2 sts=2 expandtab