You've already forked STARK
mirror of
https://github.com/MarkParker5/STARK.git
synced 2026-06-08 16:39:30 +02:00
Mark Parker
134 lines
7.7 KiB
Python
134 lines
7.7 KiB
Python
from stark.core import Pattern
|
|
from stark.core.patterns import expressions
|
|
from stark.general.localisation import Localizer
|
|
|
|
|
|
word = fr'[{expressions.alphanumerics}]*'
|
|
words = fr'[{expressions.alphanumerics}\s]*'
|
|
|
|
async def test_leading_star(get_transcription):
|
|
p = Pattern('*text')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == fr'{word}text'
|
|
assert await p.match(get_transcription('text'), Localizer())
|
|
assert await p.match(get_transcription('aaatext'), Localizer())
|
|
assert (await p.match(get_transcription('bbb aaaatext cccc'), Localizer()))[0].subtrack.text == 'aaaatext'
|
|
assert not await p.match(get_transcription('aaaaext'), Localizer())
|
|
|
|
p = Pattern('Some *text here')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == fr'Some {word}text here'
|
|
assert await p.match(get_transcription('Some text here'), Localizer())
|
|
assert await p.match(get_transcription('Some aaatext here'), Localizer())
|
|
assert (await p.match(get_transcription('bbb Some aaatext here cccc'), Localizer()))[0].subtrack.text == 'Some aaatext here'
|
|
assert not await p.match(get_transcription('Some aaatext'), Localizer())
|
|
|
|
async def test_trailing_star(get_transcription):
|
|
p = Pattern('text*')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == fr'text{word}'
|
|
assert await p.match(get_transcription('text'), Localizer())
|
|
assert await p.match(get_transcription('textaaa'), Localizer())
|
|
assert (await p.match(get_transcription('bbb textaaa cccc'), Localizer()))[0].subtrack.text == 'textaaa'
|
|
|
|
p = Pattern('Some text* here')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == fr'Some text{word} here'
|
|
assert await p.match(get_transcription('Some text here'), Localizer())
|
|
assert await p.match(get_transcription('Some textaaa here'), Localizer())
|
|
assert (await p.match(get_transcription('bbb Some textaaa here cccc'), Localizer()))[0].subtrack.text == 'Some textaaa here'
|
|
assert not await p.match(get_transcription('Some textaaa '), Localizer())
|
|
|
|
async def test_middle_star(get_transcription):
|
|
p = Pattern('te*xt')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == fr'te{word}xt'
|
|
assert await p.match(get_transcription('text'), Localizer())
|
|
assert await p.match(get_transcription('teaaaaaxt'), Localizer())
|
|
assert (await p.match(get_transcription('bbb teaaaaaxt cccc'), Localizer()))[0].subtrack.text == 'teaaaaaxt'
|
|
|
|
p = Pattern('Some te*xt here')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == fr'Some te{word}xt here'
|
|
assert await p.match(get_transcription('Some text here'), Localizer())
|
|
assert await p.match(get_transcription('Some teaaaaaxt here'), Localizer())
|
|
assert (await p.match(get_transcription('bbb Some teaaeaaaxt here cccc'), Localizer()))[0].subtrack.text == 'Some teaaeaaaxt here'
|
|
assert not await p.match(get_transcription('Some teaaaaaxt'), Localizer())
|
|
|
|
async def test_double_star(get_transcription):
|
|
p = Pattern('**')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == fr'{words}'
|
|
assert (await p.match(get_transcription('bbb teaaaaaxt cccc'), Localizer()))[0].subtrack.text == 'bbb teaaaaaxt cccc'
|
|
|
|
p = Pattern('Some ** here')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == fr'Some {words} here'
|
|
assert await p.match(get_transcription('Some text here'), Localizer())
|
|
assert await p.match(get_transcription('Some lorem ipsum dolor here'), Localizer())
|
|
assert (await p.match(get_transcription('bbb Some lorem ipsum dolor here cccc'), Localizer()))[0].subtrack.text == 'Some lorem ipsum dolor here'
|
|
|
|
async def test_one_of(get_transcription):
|
|
p = Pattern('(foo|bar)')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == r'(?:foo|bar)'
|
|
assert await p.match(get_transcription('foo'), Localizer())
|
|
assert await p.match(get_transcription('bar'), Localizer())
|
|
assert (await p.match(get_transcription('bbb foo cccc'), Localizer()))[0].subtrack.text == 'foo'
|
|
assert (await p.match(get_transcription('bbb bar cccc'), Localizer()))[0].subtrack.text == 'bar'
|
|
|
|
p = Pattern('Some (foo|bar) here')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == r'Some (?:foo|bar) here'
|
|
assert await p.match(get_transcription('Some foo here'), Localizer())
|
|
assert await p.match(get_transcription('Some bar here'), Localizer())
|
|
assert (await p.match(get_transcription('bbb Some foo here cccc'), Localizer()))[0].subtrack.text == 'Some foo here'
|
|
assert (await p.match(get_transcription('bbb Some bar here cccc'), Localizer()))[0].subtrack.text == 'Some bar here'
|
|
assert not await p.match(get_transcription('Some foo'), Localizer())
|
|
|
|
async def test_optional_one_of(get_transcription):
|
|
p = Pattern('(foo|bar)?')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == r'(?:foo|bar)?'
|
|
assert await p.match(get_transcription('foo'), Localizer())
|
|
assert await p.match(get_transcription('bar'), Localizer())
|
|
assert not await p.match(get_transcription(''), Localizer())
|
|
assert not await p.match(get_transcription('bbb cccc'), Localizer())
|
|
assert (await p.match(get_transcription('bbb foo cccc'), Localizer()))[0].subtrack.text == 'foo'
|
|
assert (await p.match(get_transcription('bbb bar cccc'), Localizer()))[0].subtrack.text == 'bar'
|
|
|
|
p = Pattern('Some (foo|bar)? here')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == r'Some (?:foo|bar)? here'
|
|
assert await p.match(get_transcription('Some foo here'), Localizer())
|
|
assert await p.match(get_transcription('Some bar here'), Localizer())
|
|
assert await p.match(get_transcription('Some here'), Localizer())
|
|
assert (await p.match(get_transcription('bbb Some foo here cccc'), Localizer()))[0].subtrack.text == 'Some foo here'
|
|
assert (await p.match(get_transcription('bbb Some bar here cccc'), Localizer()))[0].subtrack.text == 'Some bar here'
|
|
assert (await p.match(get_transcription('bbb Some here cccc'), Localizer()))[0].subtrack.text == 'Some here'
|
|
|
|
# assert Pattern('[foo|bar]').compiled == Pattern('(foo|bar)?').compiled
|
|
|
|
async def test_one_or_more_of(get_transcription):
|
|
p = Pattern('{foo|bar}')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == r'(?:(?:foo|bar)\s?)+'
|
|
assert await p.match(get_transcription('foo'), Localizer())
|
|
assert await p.match(get_transcription('bar'), Localizer())
|
|
assert not await p.match(get_transcription(''), Localizer())
|
|
assert (await p.match(get_transcription('bbb foo cccc'), Localizer()))[0].subtrack.text == 'foo'
|
|
assert (await p.match(get_transcription('bbb bar cccc'), Localizer()))[0].subtrack.text == 'bar'
|
|
assert (await p.match(get_transcription('bbb foo bar cccc'), Localizer()))[0].subtrack.text == 'foo bar'
|
|
assert not await p.match(get_transcription('bbb cccc'), Localizer())
|
|
|
|
p = Pattern('Some {foo|bar} here')
|
|
p.get_compiled('en', Localizer())
|
|
# assert p._compiled['en'] == r'Some (?:(?:foo|bar)\s?)+ here'
|
|
assert await p.match(get_transcription('Some foo here'), Localizer())
|
|
assert await p.match(get_transcription('Some bar here'), Localizer())
|
|
assert not await p.match(get_transcription('Some here'), Localizer())
|
|
assert (await p.match(get_transcription('bbb Some foo here cccc'), Localizer()))[0].subtrack.text == 'Some foo here'
|
|
assert (await p.match(get_transcription('bbb Some bar here cccc'), Localizer()))[0].subtrack.text == 'Some bar here'
|
|
assert (await p.match(get_transcription('bbb Some foo bar here cccc'), Localizer()))[0].subtrack.text == 'Some foo bar here'
|
|
assert not await p.match(get_transcription('Some foo'), Localizer())
|