1d176050a7
This implements %regex_sub built-in function to enhance string-processing capabilities, allowing users to perform substitutes like: - removing non-ascii characters - replacing subsequent whitespace characters with a single whitespace Thanks @tbabej !
61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
import pytest
|
|
from unit.script.conftest import single_variable_output
|
|
|
|
from ytdl_sub.script.script import Script
|
|
|
|
|
|
class TestNumericFunctions:
|
|
@pytest.mark.parametrize(
|
|
"values, expected_output",
|
|
[
|
|
("'ow', 'lower'", []),
|
|
("'.*ow.*', 'lower'", ["lower"]),
|
|
("'.*(ow).*', 'lower'", ["lower", "ow"]),
|
|
("'(.*)(ow)(.*)', 'lower'", ["lower", "l", "ow", "er"]),
|
|
],
|
|
)
|
|
def test_regex_match(self, values: str, expected_output: str):
|
|
output = single_variable_output(f"{{%regex_match({values})}}")
|
|
assert output == expected_output
|
|
|
|
@pytest.mark.parametrize(
|
|
"values, expected_output",
|
|
[
|
|
("'ow', 'lower'", ["lower"]),
|
|
("'.*ow.*', 'lower'", ["lower"]),
|
|
("'.*(ow).*', 'lower'", ["lower", "ow"]),
|
|
("'(.*)(ow)(.*)', 'lower'", ["lower", "l", "ow", "er"]),
|
|
],
|
|
)
|
|
def test_regex_search(self, values: str, expected_output: str):
|
|
output = single_variable_output(f"{{%regex_search({values})}}")
|
|
assert output == expected_output
|
|
|
|
@pytest.mark.parametrize(
|
|
"values, expected_output",
|
|
[
|
|
("'ow', 'lower'", []),
|
|
("'.*ow.*', 'lower'", ["lower"]),
|
|
("'.*(ow).*', 'lower'", ["lower", "ow"]),
|
|
("'(.*)(ow)(.*)', 'lower'", ["lower", "l", "ow", "er"]),
|
|
],
|
|
)
|
|
def test_regex_fullmatch(self, values: str, expected_output: str):
|
|
output = single_variable_output(f"{{%regex_fullmatch({values})}}")
|
|
assert output == expected_output
|
|
|
|
@pytest.mark.parametrize(
|
|
"values, expected_output",
|
|
[
|
|
("'[^A-Za-z0-9 ]', '', 'This title is AWESOME!!'", "This title is AWESOME"),
|
|
("'\s+', '_', 'Consolidate spaces'", "Consolidate_spaces"),
|
|
(
|
|
"'(words) are (reordered)', '\\2 are \\1', 'Oh words are reordered'",
|
|
"Oh reordered are words",
|
|
),
|
|
("'MATCH', '', 'matcha is great'", "matcha is great"),
|
|
],
|
|
)
|
|
def test_regex_sub(self, values: str, expected_output: str):
|
|
output = single_variable_output(f"{{%regex_sub({values})}}")
|
|
assert output == expected_output
|