[enh] testcases added

This commit is contained in:
asciimoo 2012-07-04 12:02:46 +02:00
parent ae0edfdf9c
commit f7d5f47a9b
1 changed files with 26 additions and 0 deletions

26
tests.py Normal file
View File

@ -0,0 +1,26 @@
from exrex import generate, count
RS = {'[ab][cd]': ['ac', 'ad', 'bc', 'bd']
,'[12]{1,2}': ['1', '2', '11', '12', '21', '22']
,'((hai){2}|world)!': ['haihai!', 'world!']
,'[ab]{1,3}': ['a', 'b', 'aa', 'ab', 'ba', 'bb', 'aaa', 'aab', 'aba', 'abb', 'baa', 'bab', 'bba', 'bbb']
,'\d': map(str, range(0, 10))
}
def gen_test():
for regex, result in RS.items():
assert list(generate(regex)) == result
def count_test():
for regex, result in RS.items():
if not count(regex) == len(result):
print '[LENERR] "%s" - %d:%d' % (regex, count(regex), len(result))
if __name__ == '__main__':
gen_test()
print '[!] generation test passed'
count_test()