---
source: src/main.rs
expression: compiled
input_file: test-data/lua5.3-tests/pm.lua
---
print('testing pattern matching');
local fn checkerror(msg, f, ...) {
local s, err = pcall(f, ...)
assert(!s && string.find(err, msg));
}
global fn f(s, p) {
local i, e = string.find(s, p)
if i {
return string.sub(s, i, e)
}
}
a, b = string.find('', '')
assert(a == 1 && b == 0);
a, b = string.find('alo', '')
assert(a == 1 && b == 0);
a, b = string.find('a\0o a\0o a\0o', 'a', 1)
assert(a == 1 && b == 1);
a, b = string.find('a\0o a\0o a\0o', 'a\0o', 2)
assert(a == 5 && b == 7);
a, b = string.find('a\0o a\0o a\0o', 'a\0o', 9)
assert(a == 9 && b == 11);
a, b = string.find('a\0a\0a\0a\0\0ab', '\0ab', 2)
assert(a == 9 && b == 11);
a, b = string.find('a\0a\0a\0a\0\0ab', 'b')
assert(a == 11 && b == 11);
assert(string.find('a\0a\0a\0a\0\0ab', 'b\0') == nil);
assert(string.find('', '\0') == nil);
assert(string.find('alo123alo', '12') == 4);
assert(string.find('alo123alo', '^12') == nil);
assert(string.match("aaab", ".*b") == "aaab");
assert(string.match("aaa", ".*a") == "aaa");
assert(string.match("b", ".*b") == "b");
assert(string.match("aaab", ".+b") == "aaab");
assert(string.match("aaa", ".+a") == "aaa");
assert(!string.match("b", ".+b"));
assert(string.match("aaab", ".?b") == "ab");
assert(string.match("aaa", ".?a") == "aa");
assert(string.match("b", ".?b") == "b");
assert(f('aloALO', '%l*') == 'alo');
assert(f('aLo_ALO', '%a*') == 'aLo');
assert(f(" \n\r*&\n\r xuxu \n\n", "%g%g%g+") == "xuxu");
assert(f('aaab', 'a*') == 'aaa');
assert(f('aaa', '^.*$') == 'aaa');
assert(f('aaa', 'b*') == '');
assert(f('aaa', 'ab*a') == 'aa');
assert(f('aba', 'ab*a') == 'aba');
assert(f('aaab', 'a+') == 'aaa');
assert(f('aaa', '^.+$') == 'aaa');
assert(f('aaa', 'b+') == nil);
assert(f('aaa', 'ab+a') == nil);
assert(f('aba', 'ab+a') == 'aba');
assert(f('a$a', '.$') == 'a');
assert(f('a$a', '.%$') == 'a$');
assert(f('a$a', '.$.') == 'a$a');
assert(f('a$a', '$$') == nil);
assert(f('a$b', 'a$') == nil);
assert(f('a$a', '$') == '');
assert(f('', 'b*') == '');
assert(f('aaa', 'bb*') == nil);
assert(f('aaab', 'a-') == '');
assert(f('aaa', '^.-$') == 'aaa');
assert(f('aabaaabaaabaaaba', 'b.*b') == 'baaabaaabaaab');
assert(f('aabaaabaaabaaaba', 'b.-b') == 'baaab');
assert(f('alo xo', '.o$') == 'xo');
assert(f(' \n isto é assim', '%S%S*') == 'isto');
assert(f(' \n isto é assim', '%S*$') == 'assim');
assert(f(' \n isto é assim', '[a-z]*$') == 'assim');
assert(f('um caracter ? extra', '[^%sa-z]') == '?');
assert(f('', 'a?') == '');
assert(f('á', 'á?') == 'á');
assert(f('ábl', 'á?b?l?') == 'ábl');
assert(f(' ábl', 'á?b?l?') == '');
assert(f('aa', '^aa?a?a') == 'aa');
assert(f(']]]áb', '[^]]') == 'á');
assert(f("0alo alo", "%x*") == "0a");
assert(f("alo alo", "%C+") == "alo alo");
print('+');
global fn f1(s, p) {
p = string.gsub(p, "%%([0-9])", fn (s) {
return "%" .. (tonumber(s) + 1)
})
p = string.gsub(p, "^(^?)", "%1()", 1)
p = string.gsub(p, "($?)$", "()%1", 1)
local t = {
string.match(s, p)
}
return string.sub(s, t[(1)], t[(#t)] - 1)
}
assert(f1('alo alx 123 b\0o b\0o', '(..*) %1') == "b\0o b\0o");
assert(f1('axz123= 4= 4 34', '(.+)=(.*)=%2 %1') == '3= 4= 4 3');
assert(f1('=======', '^(=*)=%1$') == '=======');
assert(string.match('==========', '^([=]*)=%1$') == nil);
local fn range(i, j) {
if i <= j {
return i, range(i + 1, j)
}
}
local abc = string.char(range(0, 255))
assert(string.len(abc) == 256);
global fn strset(p) {
local res = {
s = ''
}
string.gsub(abc, p, fn (c) {
res.s = res.s .. c
});
return res.s
}
assert(string.len(strset('[\200-\210]')) == 11);
assert(strset('[a-z]') == "abcdefghijklmnopqrstuvwxyz");
assert(strset('[a-z%d]') == strset('[%da-uu-z]'));
assert(strset('[a-]') == "-a");
assert(strset('[^%W]') == strset('[%w]'));
assert(strset('[]%%]') == '%]');
assert(strset('[a%-z]') == '-az');
assert(strset('[%^%[%-a%]%-b]') == '-[]^ab');
assert(strset('%Z') == strset('[\1-\255]'));
assert(strset('.') == strset('[\1-\255%z]'));
print('+');
assert(string.match("alo xyzK", "(%w+)K") == "xyz");
assert(string.match("254 K", "(%d*)K") == "");
assert(string.match("alo ", "(%w*)$") == "");
assert(string.match("alo ", "(%w+)$") == nil);
assert(string.find("(álo)", "%(á") == 1);
local a, b, c, d, e = string.match("âlo alo", "^(((.).).* (%w*))$")
assert(a == 'âlo alo' && b == 'âl' && c == 'â' && d == 'alo' && e == nil);
a, b, c, d = string.match('0123456789', '(.+(.?)())')
assert(a == '0123456789' && b == '' && c == 11 && d == nil);
print('+');
assert(string.gsub('ülo ülo', 'ü', 'x') == 'xlo xlo');
assert(string.gsub('alo úlo ', ' +$', '') == 'alo úlo');
assert(string.gsub(' alo alo ', '^%s*(.-)%s*$', '%1') == 'alo alo');
assert(string.gsub('alo alo \n 123\n ', '%s+', ' ') == 'alo alo 123 ');
t = "abç d"
a, b = string.gsub(t, '(.)', '%1@')
assert('@' .. a == string.gsub(t, '', '@') && b == 5);
a, b = string.gsub('abçd', '(.)', '%0@', 2)
assert(a == 'a@b@çd' && b == 2);
assert(string.gsub('alo alo', '()[al]', '%1') == '12o 56o');
assert(string.gsub("abc=xyz", "(%w*)(%p)(%w+)", "%3%2%1-%0") == "xyz=abc-abc=xyz");
assert(string.gsub("abc", "%w", "%1%0") == "aabbcc");
assert(string.gsub("abc", "%w+", "%0%1") == "abcabc");
assert(string.gsub('áéí', '$', '\0óú') == 'áéí\0óú');
assert(string.gsub('', '^', 'r') == 'r');
assert(string.gsub('', '$', 'r') == 'r');
print('+');
{
assert(string.gsub("a b cd", " *", "-") == "-a-b-c-d-");
local res = ""
local sub = "a \nbc\t\td"
local i = 1
for p, e with string.gmatch(sub, "()%s*()") {
res = res .. string.sub(sub, i, p - 1) .. "-"
i = e
}
assert(res == "-a-b-c-d-");
}
assert(string.gsub("um (dois) tres (quatro)", "(%(%w+%))", string.upper) == "um (DOIS) tres (QUATRO)");
{
local fn setglobal(n, v) {
rawset(_G, n, v);
}
string.gsub("a=roberto,roberto=a", "(%w+)=(%w%w*)", setglobal);
assert(_G.a == "roberto" && _G.roberto == "a");
}
global fn f(a, b) {
return string.gsub(a, '.', b)
}
assert(string.gsub("trocar tudo em |teste|b| é |beleza|al|", "|([^|]*)|([^|]*)|", f) == "trocar tudo em bbbbb é alalalalalal");
local fn dostring(s) {
return load(s, "")() || ""
}
assert(string.gsub("alo $a='x'$ novamente $return a$", "$([^$]*)%$", dostring) == "alo novamente x");
x = string.gsub("$x=string.gsub('alo', '.', string.upper)$ assim vai para $return x$", "$([^$]*)%$", dostring)
assert(x == ' assim vai para ALO');
t = {}
s = 'a alo jose joao'
r = string.gsub(s, '()(%w+)()', fn (a, w, b) {
assert(string.len(w) == b - a);
t[(a)] = b - a
})
assert(s == r && t[(1)] == 1 && t[(3)] == 3 && t[(7)] == 4 && t[(13)] == 4);
global fn isbalanced(s) {
return string.find(string.gsub(s, "%b()", ""), "[()]") == nil
}
assert(isbalanced("(9 ((8))(\0) 7) \0\0 a b ()(c)() a"));
assert(!isbalanced("(9 ((8) 7) a b (\0 c) a"));
assert(string.gsub("alo 'oi' alo", "%b''", '"') == 'alo " alo');
local t = {
"apple",
"orange",
"lime",
n = 0
}
assert(string.gsub("x and x and x", "x", fn () {
t.n = t.n + 1
return t[(t.n)]
}) == "apple and orange and lime");
t = {
n = 0
}
string.gsub("first second word", "%w%w*", fn (w) {
t.n = t.n + 1
t[(t.n)] = w
});
assert(t[(1)] == "first" && t[(2)] == "second" && t[(3)] == "word" && t.n == 3);
t = {
n = 0
}
assert(string.gsub("first second word", "%w+", fn (w) {
t.n = t.n + 1
t[(t.n)] = w
}, 2) == "first second word");
assert(t[(1)] == "first" && t[(2)] == "second" && t[(3)] == nil);
checkerror("invalid replacement value %(a table%)", string.gsub, "alo", ".", {
a = {}
});
checkerror("invalid capture index %%2", string.gsub, "alo", ".", "%2");
checkerror("invalid capture index %%0", string.gsub, "alo", "(%0)", "a");
checkerror("invalid capture index %%1", string.gsub, "alo", "(%1)", "a");
checkerror("invalid use of '%%'", string.gsub, "alo", ".", "%x");
{
local fn f(size) {
local s = string.rep("a", size)
local p = string.rep(".?", size)
return pcall(string.match, s, p)
}
local r, m = f(80)
assert(r && #m == 80);
r, m = f(200000)
assert(!r && string.find(m, "too complex"));
}
if !_soft {
print("big strings");
local a = string.rep('a', 300000)
assert(string.find(a, '^a*.?$'));
assert(!string.find(a, '^a*.?b$'));
assert(string.find(a, '^a-.?$'));
a = string.rep('a', 10000) .. string.rep('b', 10000)
assert(!pcall(string.gsub, a, 'b'));
}
global fn rev(s) {
return string.gsub(s, "(.)(.+)", fn (c, s1) {
return rev(s1) .. c
})
}
local x = "abcdef"
assert(rev(rev(x)) == x);
assert(string.gsub("alo alo", ".", {}) == "alo alo");
assert(string.gsub("alo alo", "(.)", {
a = "AA",
l = ""
}) == "AAo AAo");
assert(string.gsub("alo alo", "(.).", {
a = "AA",
l = "K"
}) == "AAo AAo");
assert(string.gsub("alo alo", "((.)(.?))", {
al = "AA",
o = false
}) == "AAo AAo");
assert(string.gsub("alo alo", "().", {
'x',
'yy',
'zzz'
}) == "xyyzzz alo");
t = {}
setmetatable(t, {
__index = fn (t, s) {
return string.upper(s)
}
});
assert(string.gsub("a alo b hi", "%w%w+", t) == "a ALO b HI");
local a = 0
for i with string.gmatch('abcde', '()') {
assert(i == a + 1);
a = i
}
assert(a == 6);
t = {
n = 0
}
for w with string.gmatch("first second word", "%w+") {
t.n = t.n + 1
t[(t.n)] = w
}
assert(t[(1)] == "first" && t[(2)] == "second" && t[(3)] == "word");
t = {
3,
6,
9
}
for i with string.gmatch("xuxx uu ppar r", "()(.)%2") {
assert(i == table.remove(t, 1));
}
assert(#t == 0);
t = {}
for i, j with string.gmatch("13 14 10 = 11, 15= 16, 22=23", "(%d+)%s*=%s*(%d+)") {
t[(tonumber(i))] = tonumber(j)
}
a = 0
for k, v with pairs(t) {
assert(k + 1 == v + 0);
a = a + 1
}
assert(a == 3);
assert(string.gsub("aaa aa a aaa a", "%f[%w]a", "x") == "xaa xa x xaa x");
assert(string.gsub("[[]] [][] [[[[", "%f[[].", "x") == "x[]] x]x] x[[[");
assert(string.gsub("01abc45de3", "%f[%d]", ".") == ".01abc.45de.3");
assert(string.gsub("01abc45 de3x", "%f[%D]%w", ".") == "01.bc45 de3.");
assert(string.gsub("function", "%f[\1-\255]%w", ".") == ".unction");
assert(string.gsub("function", "%f[^\1-\255]", ".") == "function.");
assert(string.find("a", "%f[a]") == 1);
assert(string.find("a", "%f[^%z]") == 1);
assert(string.find("a", "%f[^%l]") == 2);
assert(string.find("aba", "%f[a%z]") == 3);
assert(string.find("aba", "%f[%z]") == 4);
assert(!string.find("aba", "%f[%l%z]"));
assert(!string.find("aba", "%f[^%l%z]"));
local i, e = string.find(" alo aalo allo", "%f[%S].-%f[%s].-%f[%S]")
assert(i == 2 && e == 5);
local k = string.match(" alo aalo allo", "%f[%S](.-%f[%s].-%f[%S])")
assert(k == 'alo ');
local a = {
1,
5,
9,
14,
17
}
for k with string.gmatch("alo alo th02 is 1hat", "()%f[%w%d]") {
assert(table.remove(a, 1) == k);
}
assert(#a == 0);
local fn malform(p, m) {
m = m || "malformed"
local r, msg = pcall(string.find, "a", p)
assert(!r && string.find(msg, m));
}
malform("(.", "unfinished capture");
malform(".)", "invalid pattern capture");
malform("[a");
malform("[]");
malform("[^]");
malform("[a%]");
malform("[a%");
malform("%b");
malform("%ba");
malform("%");
malform("%f", "missing");
assert(string.match("ab\0\1\2c", "[\0-\2]+") == "\0\1\2");
assert(string.match("ab\0\1\2c", "[\0-\0]+") == "\0");
assert(string.find("b$a", "$\0?") == 2);
assert(string.find("abc\0efg", "%\0") == 4);
assert(string.match("abc\0efg\0\1e\1g", "%b\0\1") == "\0efg\0\1e\1");
assert(string.match("abc\0\0\0", "%\0+") == "\0\0\0");
assert(string.match("abc\0\0\0", "%\0%\0?") == "\0\0");
assert(string.find("abc\0\0", "\0.") == 4);
assert(string.find("abcx\0\0abc\0abc", "x\0\0abc\0a.") == 4);
print('OK');