veniasilente/veekun_pokedex
1
0
Fork 0
mirror of https://github.com/veekun/pokedex.git synced 2024-08-20 18:16:34 +00:00
Code Issues Projects Releases Wiki Activity

Share session across tests

Browse source
This commit is contained in:
Petr Viktorin 2011-04-27 14:30:11 +03:00
parent d746716575
commit e4c0d0b16b
2 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
pokedex/tests/test_movesets.py
View file

@ -1,9 +1,11 @@
from pokedex.db import connect
from pokedex.util.movesets import main
result_map = {'OK': True, 'NO': False}
def test_cases():
session = connect()
for argstring in u"""
NO muk
NO beedrill rage pursuit agility endeavor toxic
@ -43,7 +45,7 @@ def test_cases():
""".strip().splitlines():
def run_test(argstring):
args = argstring.split()
assert main(args[1:]) == result_map[args[0]]
assert main(args[1:], session=session) == result_map[args[0]]
run_test.description = 'Moveset checker test: ' + argstring.strip()
yield run_test, argstring.strip()
@ -55,6 +57,6 @@ if __name__ == '__main__':
def header(str):
print
print str
cProfile.runctx("[(header(arg), f(arg)) for f, arg in test_cases()]",
cProfile.runctx("[(header(argv), f(argv)) for f, argv in test_cases()]",
globals(), locals(), filename=filename)
print 'Profile stats saved to', filename

3
pokedex/util/movesets.py
View file

@ -1289,7 +1289,7 @@ def print_result(result, moves=()):
''.join(m.name[0].lower() for m in node.moves if m not in moves),
)
def main(argv):
def main(argv, session=None):
parser = argparse.ArgumentParser(description=
'Find out if the specified moveset is valid, and provide a suggestion '
'on how to obtain it.')
@ -1328,6 +1328,7 @@ def main(argv):
if args.debug:
print 'Connecting'
if session is None:
session = connect(engine_args={'echo': args.debug > 2})
if args.debug: