micropython/tests/extmod/ure_sub.py
David Lechner 3dc324d3f1 tests: Format all Python code with black, except tests in basics subdir.
This adds the Python files in the tests/ directory to be formatted with
./tools/codeformat.py.  The basics/ subdirectory is excluded for now so we
aren't changing too much at once.

In a few places `# fmt: off`/`# fmt: on` was used where the code had
special formatting for readability or where the test was actually testing
the specific formatting.
2020-03-30 13:21:58 +11:00

63 lines
1.2 KiB
Python

try:
import ure as re
except ImportError:
try:
import re
except ImportError:
print("SKIP")
raise SystemExit
try:
re.sub
except AttributeError:
print("SKIP")
raise SystemExit
def multiply(m):
return str(int(m.group(0)) * 2)
print(re.sub("\d+", multiply, "10 20 30 40 50"))
print(re.sub("\d+", lambda m: str(int(m.group(0)) // 2), "10 20 30 40 50"))
def A():
return "A"
print(re.sub("a", A(), "aBCBABCDabcda."))
print(
re.sub(
r"def\s+([a-zA-Z_][a-zA-Z_0-9]*)\s*\(\s*\):",
"static PyObject*\npy_\\1(void){\n return;\n}\n",
"\n\ndef myfunc():\n\ndef myfunc1():\n\ndef myfunc2():",
)
)
print(
re.compile("(calzino) (blu|bianco|verde) e (scarpa) (blu|bianco|verde)").sub(
r"\g<1> colore \2 con \g<3> colore \4? ...", "calzino blu e scarpa verde"
)
)
# no matches at all
print(re.sub("a", "b", "c"))
# with maximum substitution count specified
print(re.sub("a", "b", "1a2a3a", 2))
# invalid group
try:
re.sub("(a)", "b\\2", "a")
except:
print("invalid group")
# invalid group with very large number (to test overflow in uPy)
try:
re.sub("(a)", "b\\199999999999999999999999999999999999999", "a")
except:
print("invalid group")