class TestRegex(unittest.TestCase):
def setUp(self):
"""Inicia novo objeto em todo os testes.
"""
self.regex = Regex("[A-z]")
def test_eq(self):
"""Verifica se o método __eq__ funciona.
"""
self.assertTrue(self.regex == "[A-z]")
def test_iadd(self):
"""Verifica se o método __iadd__ funciona.
"""
self.regex += "{3}"
self.assertTrue(self.regex == "[A-z]{3}")
def test_set(self):
"""Verifica se o método _set_regex funciona.
"""
self.regex._set_regex("[0-9]{5}")
self.assertTrue(self.regex == "[0-9]{5}")
def test_add(self):
"""Verifica se o método __add__ funciona.
"""
self.regex = Regex("[0-9]{5}") + "-" + Regex("[0-9]{3}")
self.assertTrue(self.regex == "[0-9]{5}-[0-9]{3}")
def test_set(self):
regex = Regex('[A-z]')
regex._set_regex('[0-9]{5}')
self.assertTrue(regex == '[0-9]{5}')
from mre import Regex
test = Regex("[A-z]")
# __eq__
assert test == "[A-z]"
# __iadd__
test += "{3}"
assert test == "[A-z]{3}"
# set
test._set_regex("[0-9]{5}")
assert test == "[0-9]{5}"
# __add__
test_cep = test + "-" + Regex("[0-9]{3}")
assert test_cep == "[0-9]{5}-[0-9]{3}"
