diff --git a/tests.py b/tests.py new file mode 100644 index 0000000..f8d9d41 --- /dev/null +++ b/tests.py @@ -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()