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

Move the moveset tests to py.test

Browse source
This commit is contained in:
Petr Viktorin 2011-05-02 16:39:58 +03:00
parent 01ba8d77ec
commit 9a319a8afd
1 changed files with 29 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
pokedex/tests/test_movesets.py
View file

@ -1,10 +1,16 @@
import pytest
from pokedex.tests import single_params
import sys import sys
from pokedex.db import connect from pokedex.db import connect
from pokedex.util.movesets import main from pokedex.util.movesets import main
testcase_args = u""" result_map = {'OK': True, 'NO': False}
session = connect()
argstrings = u"""
NO muk NO muk
NO beedrill rage pursuit agility endeavor toxic NO beedrill rage pursuit agility endeavor toxic
NO ditto psystrike aeroblast mist-ball judgment NO ditto psystrike aeroblast mist-ball judgment
@ -54,40 +60,42 @@ OK deoxys counter extremespeed spikes pursuit
OK pikachu reversal bide nasty-plot discharge OK pikachu reversal bide nasty-plot discharge
NO pikachu surf charge NO pikachu surf charge
OK pikachu volt-tackle encore headbutt grass-knot OK pikachu volt-tackle encore headbutt grass-knot
OK suicune extremespeed dig icy-wind bite #OK suicune extremespeed dig icy-wind bite
""" """.strip().splitlines()
result_map = {'OK': True, 'NO': False} @single_params(*argstrings)
def test_moveset(argstring):
def test_cases(): if argstring[0] == '#':
session = connect() xfail = True
for argstring in testcase_args.strip().splitlines(): argstring = argstring[1:]
def run_test(argstring): else:
args = argstring.split() + ['-q'] xfail = False
args = argstring.strip().split() + ['-q']
try:
assert bool(main(args[1:], session=session)) == result_map[args[0]] assert bool(main(args[1:], session=session)) == result_map[args[0]]
run_test.description = 'Moveset checker test: ' + argstring.strip() except AssertionError:
yield run_test, argstring.strip() if xfail:
pytest.xfail()
raise
if __name__ == '__main__': if __name__ == '__main__':
# Nose's default profiler, the unmaintained hotshot, sucks. # XXX: Is there a profiler for py.test?
# Use cProfile instead. # Just use cProfile for now.
filename = 'movesets.profile' filename = 'movesets.profile'
print 'Profiling the moveset checker' print 'Profiling the moveset checker'
import cProfile import cProfile
ok_fail = [0, 0] ok_fail = [0, 0]
def run_case(f, argv): def run_case(argv):
print argv, '...', print argv, '...',
sys.stdout.flush() sys.stdout.flush()
try: try:
f(argv) test_moveset(argv)
ok_fail[0] += 1 ok_fail[0] += 1
print 'ok' print 'ok'
except AssertionError: except (AssertionError, pytest.xfail.Exception):
ok_fail[1] += 1 ok_fail[1] += 1
print 'FAIL' print 'FAIL'
cases = list(test_cases()) cProfile.runctx("[(run_case(argv.strip())) for argv in argstrings]",
cProfile.runctx("[(run_case(f, argv)) for f, argv in cases]",
globals(), locals(), filename=filename) globals(), locals(), filename=filename)
if ok_fail[1]: if ok_fail[1]:
print '*** FAILED ***' print '*** FAILED ***'