mirror of
https://github.com/veekun/pokedex.git
synced 2024-08-20 18:16:34 +00:00
Modernize our use of py.test
This commit updates the tests to take advantage of some of py.test's newer features. Requires py.test 2.3 or newer. Tested with 2.3.0 and 2.5.2. Tests which were parametrized now use py.test's built-in parametrization[1]. The session and lookup objects are now implemented as fixtures[2]. The media root is a fixture as well. Fixtures are automatically passed to any function that expects them. Since the session is now created in one place, it is now possible to provide an engine URI on the command line when running py.test. Ditto for the index directory. (But the environment variables still work of course.) Slow tests are now marked as such and not run unless the --all option is given. A couple media tests are marked as xfail (expected to fail) because they are broken. [1]: http://pytest.org/latest/parametrize.html [2]: http://pytest.org/latest/fixture.html
This commit is contained in:
Andrew Ekstedt
parent
29824c73f4
commit
0094e9584c
10 changed files with 285 additions and 299 deletions
40
conftest.py
40
conftest.py
|
|
@ -1,4 +1,3 @@
|
|||
|
||||
# Configuration for the tests.
|
||||
# Use `py.test` to run the tests.
|
||||
|
||||
|
|
@ -8,14 +7,37 @@ import pytest
|
|||
import os
|
||||
|
||||
def pytest_addoption(parser):
|
||||
parser.addoption("--media-root", action="store",
|
||||
default=None,
|
||||
group = parser.getgroup("pokedex")
|
||||
group.addoption("--engine", action="store", default=None,
|
||||
help="Pokedex database URI")
|
||||
group.addoption("--index", action="store", default=None,
|
||||
help="Path to index directory")
|
||||
group.addoption("--media-root", action="store", default=None,
|
||||
help="Root for the media files (if not specified and pokedex/data/media doesn't exist, tests are skipped)")
|
||||
parser.addoption("--all", action="store_true", default=False,
|
||||
group.addoption("--all", action="store_true", default=False,
|
||||
help="Run all tests, even those that take a lot of time")
|
||||
|
||||
def pytest_generate_tests(metafunc):
|
||||
for funcargs in getattr(metafunc.function, 'funcarglist', ()):
|
||||
metafunc.addcall(funcargs=funcargs)
|
||||
for posargs in getattr(metafunc.function, 'posarglist', ()):
|
||||
metafunc.addcall(funcargs=dict(zip(metafunc.funcargnames, posargs)))
|
||||
def pytest_runtest_setup(item):
|
||||
if 'slow' in item.keywords and not item.config.getvalue('all'):
|
||||
pytest.skip("skipping slow tests")
|
||||
|
||||
@pytest.fixture(scope="module")
|
||||
def session(request):
|
||||
import pokedex.db
|
||||
engine_uri = request.config.getvalue("engine")
|
||||
return pokedex.db.connect(engine_uri)
|
||||
|
||||
@pytest.fixture(scope="module")
|
||||
def lookup(request, session):
|
||||
import pokedex.lookup
|
||||
index_dir = request.config.getvalue("index")
|
||||
return pokedex.lookup.PokedexLookup(index_dir, session)
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def media_root(request):
|
||||
media_root = request.config.getvalue("media_root")
|
||||
if not media_root:
|
||||
media_root = os.path.join(os.path.dirname(__file__), '..', 'data', 'media')
|
||||
if not os.path.isdir(media_root):
|
||||
raise pytest.skip("Media unavailable")
|
||||
return media_root
|
||||
|
|
|
|||
|
|
@ -1,45 +0,0 @@
|
|||
|
||||
import inspect
|
||||
from functools import wraps
|
||||
|
||||
# test support code
|
||||
def params(funcarglist):
|
||||
"""Basic list-of-dicts test parametrization
|
||||
|
||||
From: http://pytest.org/funcargs.html
|
||||
|
||||
Example:
|
||||
@params([dict(a=1, b=2), dict(a=3, b=4)])
|
||||
def test_lt(a, b):
|
||||
assert a < b
|
||||
"""
|
||||
def decorator(function):
|
||||
function.funcarglist = funcarglist
|
||||
return function
|
||||
return decorator
|
||||
|
||||
def positional_params(*paramlist):
|
||||
"""Magic list-of-lists parametrization
|
||||
|
||||
Example:
|
||||
@params([(1, 2), (3, 4)])
|
||||
def test_lt(a, b):
|
||||
assert a < b
|
||||
"""
|
||||
def decorator(function):
|
||||
function.posarglist = paramlist
|
||||
return function
|
||||
return decorator
|
||||
|
||||
def single_params(*paramlist):
|
||||
"""Magic list-of-lists parametrization
|
||||
|
||||
Example:
|
||||
@params('1', '2', '3', '4'])
|
||||
def test_int(k):
|
||||
assert int(k)
|
||||
"""
|
||||
def decorator(function):
|
||||
function.posarglist = [[param] for param in paramlist]
|
||||
return function
|
||||
return decorator
|
||||
|
|
@ -1,5 +1,5 @@
|
|||
|
||||
import pytest
|
||||
parametrize = pytest.mark.parametrize
|
||||
|
||||
from sqlalchemy.orm import aliased, joinedload, lazyload
|
||||
from sqlalchemy.orm.exc import NoResultFound
|
||||
|
|
@ -7,11 +7,11 @@ from sqlalchemy.sql import func
|
|||
|
||||
from pokedex.db import connect, tables, util
|
||||
|
||||
def test_encounter_slots():
|
||||
# Encounters have a version, which has a version group; encounters also
|
||||
# have an encounter_slot, which has a version group. The two version
|
||||
# groups should match, universally.
|
||||
session = connect()
|
||||
def test_encounter_slots(session):
|
||||
"""Encounters have a version, which has a version group; encounters also
|
||||
have an encounter_slot, which has a version group. The two version
|
||||
groups should match, universally.
|
||||
"""
|
||||
|
||||
version_group_a = aliased(tables.VersionGroup)
|
||||
version_group_b = aliased(tables.VersionGroup)
|
||||
|
|
@ -26,27 +26,26 @@ def test_encounter_slots():
|
|||
# Encounter slots all match the encounters they belong to
|
||||
assert sanity_q.count() == 0
|
||||
|
||||
def test_nonzero_autoincrement_ids():
|
||||
@parametrize('cls', tables.mapped_classes)
|
||||
def test_nonzero_autoincrement_ids(session, cls):
|
||||
"""Check that autoincrementing ids don't contain zeroes
|
||||
|
||||
MySQL doesn't like these, see e.g. bug #580
|
||||
"""
|
||||
if 'id' not in cls.__table__.c:
|
||||
return
|
||||
if not cls.__table__.c.id.autoincrement:
|
||||
return
|
||||
|
||||
session = connect()
|
||||
for cls in tables.mapped_classes:
|
||||
if 'id' in cls.__table__.c:
|
||||
if cls.__table__.c.id.autoincrement:
|
||||
def nonzero_id(cls):
|
||||
with pytest.raises(NoResultFound):
|
||||
try:
|
||||
util.get(session, cls, id=0)
|
||||
nonzero_id.description = "No zero id in %s" % cls.__name__
|
||||
yield nonzero_id, cls
|
||||
except NoResultFound:
|
||||
pass
|
||||
else:
|
||||
pytest.fail("No zero id in %s" % cls.__name__)
|
||||
|
||||
def test_unique_form_order():
|
||||
"""Check that tone PokemonForm.order value isn't used for more species
|
||||
"""
|
||||
|
||||
session = connect()
|
||||
def test_unique_form_order(session):
|
||||
"""Check that tone PokemonForm.order value isn't used for more species"""
|
||||
|
||||
species_by_form_order = {}
|
||||
|
||||
|
|
@ -66,12 +65,10 @@ def test_unique_form_order():
|
|||
species_by_form_order[form.order].name,
|
||||
form.species.name))
|
||||
|
||||
def test_default_forms():
|
||||
def test_default_forms(session):
|
||||
"""Check that each pokemon has one default form and each species has one
|
||||
default pokemon."""
|
||||
|
||||
session = connect()
|
||||
|
||||
q = session.query(tables.Pokemon)
|
||||
q = q.join(tables.PokemonForm)
|
||||
q = q.filter(tables.PokemonForm.is_default==True)
|
||||
|
|
@ -81,9 +78,9 @@ def test_default_forms():
|
|||
|
||||
for pokemon, num_default_forms in q:
|
||||
if num_default_forms == 0:
|
||||
raise AssertionError("pokemon %s has no default forms" % pokemon.name)
|
||||
pytest.fail("pokemon %s has no default forms" % pokemon.name)
|
||||
elif num_default_forms > 1:
|
||||
raise AssertionError("pokemon %s has %d default forms" % (pokemon.name, num_default_forms))
|
||||
pytest.fail("pokemon %s has %d default forms" % (pokemon.name, num_default_forms))
|
||||
|
||||
q = session.query(tables.PokemonSpecies)
|
||||
q = q.join(tables.Pokemon)
|
||||
|
|
@ -94,6 +91,6 @@ def test_default_forms():
|
|||
|
||||
for species, num_default_pokemon in q:
|
||||
if num_default_pokemon == 0:
|
||||
raise AssertionError("species %s has no default pokemon" % species.name)
|
||||
pytest.fail("species %s has no default pokemon" % species.name)
|
||||
elif num_default_pokemon > 1:
|
||||
raise AssertionError("species %s has %d default pokemon" % (species.name, num_default_pokemon))
|
||||
pytest.fail("species %s has %d default pokemon" % (species.name, num_default_pokemon))
|
||||
|
|
|
|||
|
|
@ -1,12 +1,11 @@
|
|||
# Encoding: UTF-8
|
||||
|
||||
from pokedex.tests import *
|
||||
import pytest
|
||||
parametrize = pytest.mark.parametrize
|
||||
|
||||
from pokedex.lookup import PokedexLookup
|
||||
|
||||
lookup = PokedexLookup()
|
||||
|
||||
@positional_params(
|
||||
@parametrize(
|
||||
('input', 'table', 'id'),
|
||||
[
|
||||
# Simple lookups
|
||||
(u'Eevee', 'pokemon_species',133),
|
||||
(u'Scratch', 'moves', 10),
|
||||
|
|
@ -31,8 +30,9 @@ lookup = PokedexLookup()
|
|||
(u'이브이', 'pokemon_species', 133),
|
||||
(u'伊布', 'pokemon_species', 133),
|
||||
(u'Evoli', 'pokemon_species', 133),
|
||||
]
|
||||
)
|
||||
def test_exact_lookup(input, table, id):
|
||||
def test_exact_lookup(lookup, input, table, id):
|
||||
results = lookup.lookup(input)
|
||||
assert len(results) == 1
|
||||
assert results[0].exact == True
|
||||
|
|
@ -42,19 +42,19 @@ def test_exact_lookup(input, table, id):
|
|||
assert row.id == id
|
||||
|
||||
|
||||
def test_id_lookup():
|
||||
def test_id_lookup(lookup):
|
||||
results = lookup.lookup(u'1')
|
||||
assert len(results) >= 5
|
||||
assert all(result.object.id == 1 for result in results)
|
||||
|
||||
|
||||
def test_multi_lookup():
|
||||
def test_multi_lookup(lookup):
|
||||
results = lookup.lookup(u'Metronome')
|
||||
assert len(results) == 2
|
||||
assert results[0].exact
|
||||
|
||||
|
||||
def test_type_lookup():
|
||||
def test_type_lookup(lookup):
|
||||
results = lookup.lookup(u'pokemon:1')
|
||||
assert results[0].object.__tablename__ == 'pokemon_species'
|
||||
assert len(results) == 1
|
||||
|
|
@ -64,7 +64,7 @@ def test_type_lookup():
|
|||
assert results[0].object.name == u'Bulbasaur'
|
||||
|
||||
|
||||
def test_language_lookup():
|
||||
def test_language_lookup(lookup):
|
||||
# There are two objects named "charge": the move Charge, and the move
|
||||
# Tackle, which is called "Charge" in French.
|
||||
results = lookup.lookup(u'charge')
|
||||
|
|
@ -85,7 +85,9 @@ def test_language_lookup():
|
|||
assert results[0].object.name, u'Tackle'
|
||||
|
||||
|
||||
@positional_params(
|
||||
@parametrize(
|
||||
('misspelling', 'name'),
|
||||
[
|
||||
# Regular English names
|
||||
(u'chamander', u'Charmander'),
|
||||
(u'pokeball', u'Poké Ball'),
|
||||
|
|
@ -97,51 +99,58 @@ def test_language_lookup():
|
|||
# Sufficiently long foreign names
|
||||
(u'カクレオ', u'Kecleon'),
|
||||
(u'Yamikrasu', u'Murkrow'),
|
||||
]
|
||||
)
|
||||
def test_fuzzy_lookup(misspelling, name):
|
||||
def test_fuzzy_lookup(lookup, misspelling, name):
|
||||
results = lookup.lookup(misspelling)
|
||||
first_result = results[0]
|
||||
assert first_result.object.name == name
|
||||
|
||||
|
||||
def test_nidoran():
|
||||
def test_nidoran(lookup):
|
||||
results = lookup.lookup(u'Nidoran')
|
||||
top_names = [result.object.name for result in results[0:2]]
|
||||
assert u'Nidoran♂' in top_names
|
||||
assert u'Nidoran♀' in top_names
|
||||
|
||||
|
||||
@positional_params(
|
||||
@parametrize(
|
||||
('wildcard', 'name'),
|
||||
[
|
||||
(u'pokemon:*meleon', u'Charmeleon'),
|
||||
(u'item:master*', u'Master Ball'),
|
||||
(u'ee?ee', u'Eevee'),
|
||||
]
|
||||
)
|
||||
def test_wildcard_lookup(wildcard, name):
|
||||
def test_wildcard_lookup(lookup, wildcard, name):
|
||||
results = lookup.lookup(wildcard)
|
||||
first_result = results[0]
|
||||
assert first_result.object.name == name
|
||||
|
||||
|
||||
def test_bare_random():
|
||||
def test_bare_random(lookup):
|
||||
for i in range(5):
|
||||
results = lookup.lookup(u'random')
|
||||
assert len(results) == 1
|
||||
|
||||
|
||||
@positional_params(
|
||||
[u'pokemon_species'],
|
||||
[u'moves'],
|
||||
[u'items'],
|
||||
[u'abilities'],
|
||||
[u'types'],
|
||||
@parametrize(
|
||||
'table_name',
|
||||
[
|
||||
u'pokemon_species',
|
||||
u'moves',
|
||||
u'items',
|
||||
u'abilities',
|
||||
u'types'
|
||||
]
|
||||
)
|
||||
def test_qualified_random(table_name):
|
||||
def test_qualified_random(lookup, table_name):
|
||||
results = lookup.lookup(u'random', valid_types=[table_name])
|
||||
assert len(results) == 1
|
||||
assert results[0].object.__tablename__ == table_name
|
||||
|
||||
|
||||
def test_crash_empty_prefix():
|
||||
def test_crash_empty_prefix(lookup):
|
||||
"""Searching for ':foo' used to crash, augh!"""
|
||||
results = lookup.lookup(u':Eevee')
|
||||
assert results[0].object.name == u'Eevee'
|
||||
|
|
|
|||
|
|
@ -10,76 +10,62 @@ import pytest
|
|||
import os
|
||||
import re
|
||||
|
||||
from pokedex.db import tables, connect
|
||||
from pokedex.db import tables
|
||||
from pokedex.util import media
|
||||
|
||||
def pytest_addoption(parser):
|
||||
group = parser.getgroup("pokedex")
|
||||
group.addoption("--media-root", dest="media_root", action="store", default=None,
|
||||
help="path to pokedex-media")
|
||||
|
||||
def pytest_funcarg__root(request):
|
||||
root = request.config.option.media_root
|
||||
if not root:
|
||||
root = os.path.join(os.path.dirname(__file__), *'../data/media'.split('/'))
|
||||
if not media.BaseMedia(root).available:
|
||||
raise pytest.skip("Media unavailable")
|
||||
return root
|
||||
|
||||
session = connect()
|
||||
|
||||
path_re = re.compile('^[-a-z0-9./]*$')
|
||||
|
||||
def test_totodile(root):
|
||||
def test_totodile(session, media_root):
|
||||
"""Totodile's female sprite -- same as male"""
|
||||
totodile = session.query(tables.PokemonSpecies).filter_by(identifier=u'totodile').one()
|
||||
accessor = media.PokemonSpeciesMedia(root, totodile)
|
||||
accessor = media.PokemonSpeciesMedia(media_root, totodile)
|
||||
assert accessor.sprite() == accessor.sprite(female=True)
|
||||
|
||||
def test_chimecho(root):
|
||||
def test_chimecho(session, media_root):
|
||||
"""Chimecho's Platinum female backsprite -- diffeent from male"""
|
||||
chimecho = session.query(tables.PokemonSpecies).filter_by(identifier=u'chimecho').one()
|
||||
accessor = media.PokemonSpeciesMedia(root, chimecho)
|
||||
accessor = media.PokemonSpeciesMedia(media_root, chimecho)
|
||||
male = accessor.sprite('platinum', back=True, frame=2)
|
||||
female = accessor.sprite('platinum', back=True, female=True, frame=2)
|
||||
assert male != female
|
||||
|
||||
def test_venonat(root):
|
||||
def test_venonat(session, media_root):
|
||||
"""Venonat's shiny Yellow sprite -- same as non-shiny"""
|
||||
venonat = session.query(tables.PokemonSpecies).filter_by(identifier=u'venonat').one()
|
||||
accessor = media.PokemonSpeciesMedia(root, venonat)
|
||||
accessor = media.PokemonSpeciesMedia(media_root, venonat)
|
||||
assert accessor.sprite('yellow') == accessor.sprite('yellow', shiny=True)
|
||||
|
||||
def test_arceus_icon(root):
|
||||
"""Arceus fire-form icon -- same as base icon"""
|
||||
@pytest.mark.xfail
|
||||
def test_arceus_icon(session, media_root):
|
||||
"""Arceus normal-form icon -- same as base icon"""
|
||||
arceus = session.query(tables.PokemonSpecies).filter_by(identifier=u'arceus').one()
|
||||
accessor = media.PokemonSpeciesMedia(root, arceus)
|
||||
fire_arceus = [f for f in arceus.forms if f.form_identifier == 'fire'][0]
|
||||
fire_accessor = media.PokemonFormMedia(root, fire_arceus)
|
||||
assert accessor.icon() == fire_accessor.icon()
|
||||
accessor = media.PokemonSpeciesMedia(media_root, arceus)
|
||||
normal_arceus = [f for f in arceus.forms if f.form_identifier == 'normal'][0]
|
||||
normal_accessor = media.PokemonFormMedia(media_root, normal_arceus)
|
||||
assert accessor.icon() == normal_accessor.icon()
|
||||
|
||||
def test_strict_castform(root):
|
||||
def test_strict_castform(session, media_root):
|
||||
"""Castform rainy form overworld with strict -- unavailable"""
|
||||
with pytest.raises(ValueError):
|
||||
castform = session.query(tables.PokemonSpecies).filter_by(identifier=u'castform').first()
|
||||
rainy_castform = [f for f in castform.forms if f.form_identifier == 'rainy'][0]
|
||||
print rainy_castform
|
||||
rainy_castform = media.PokemonFormMedia(root, rainy_castform)
|
||||
rainy_castform = media.PokemonFormMedia(media_root, rainy_castform)
|
||||
rainy_castform.overworld('up', strict=True)
|
||||
|
||||
def test_strict_exeggcute(root):
|
||||
def test_strict_exeggcute(session, media_root):
|
||||
"""Exeggcutes's female backsprite, with strict -- unavailable"""
|
||||
with pytest.raises(ValueError):
|
||||
exeggcute = session.query(tables.PokemonSpecies).filter_by(identifier=u'exeggcute').one()
|
||||
accessor = media.PokemonSpeciesMedia(root, exeggcute)
|
||||
accessor = media.PokemonSpeciesMedia(media_root, exeggcute)
|
||||
accessor.sprite(female=True, strict=True)
|
||||
|
||||
|
||||
|
||||
def get_all_filenames(root):
|
||||
def get_all_filenames(media_root):
|
||||
all_filenames = set()
|
||||
|
||||
for dirpath, dirnames, filenames in os.walk(root):
|
||||
for dirpath, dirnames, filenames in os.walk(media_root):
|
||||
dirnames[:] = [dirname for dirname in dirnames if dirname != '.git']
|
||||
for filename in filenames:
|
||||
path = os.path.join(dirpath, filename)
|
||||
|
|
@ -109,8 +95,9 @@ def hit(filenames, method, *args, **kwargs):
|
|||
pass
|
||||
return True
|
||||
|
||||
@pytest.mark.skipif("not config.getvalue('all')", reason='`--all` not specified')
|
||||
def test_get_everything(root, pytestconfig):
|
||||
@pytest.mark.slow
|
||||
@pytest.mark.xfail
|
||||
def test_get_everything(session, media_root):
|
||||
"""
|
||||
For every the accessor method, loop over the Cartesian products of all
|
||||
possible values for its arguments.
|
||||
|
|
@ -119,7 +106,6 @@ def test_get_everything(root, pytestconfig):
|
|||
|
||||
Well, there are exceptions of course.
|
||||
"""
|
||||
assert pytestconfig.getvalue('all')
|
||||
|
||||
versions = list(session.query(tables.Version).all())
|
||||
versions.append('red-green')
|
||||
|
|
@ -130,28 +116,28 @@ def test_get_everything(root, pytestconfig):
|
|||
|
||||
black = session.query(tables.Version).filter_by(identifier=u'black').one()
|
||||
|
||||
filenames = get_all_filenames(root)
|
||||
filenames = get_all_filenames(media_root)
|
||||
|
||||
# Some small stuff first
|
||||
|
||||
for damage_class in session.query(tables.MoveDamageClass).all():
|
||||
assert hit(filenames, media.DamageClassMedia(root, damage_class).icon)
|
||||
assert hit(filenames, media.DamageClassMedia(media_root, damage_class).icon)
|
||||
|
||||
for habitat in session.query(tables.PokemonHabitat).all():
|
||||
assert hit(filenames, media.HabitatMedia(root, habitat).icon)
|
||||
assert hit(filenames, media.HabitatMedia(media_root, habitat).icon)
|
||||
|
||||
for shape in session.query(tables.PokemonShape).all():
|
||||
assert hit(filenames, media.ShapeMedia(root, shape).icon)
|
||||
assert hit(filenames, media.ShapeMedia(media_root, shape).icon)
|
||||
|
||||
for item_pocket in session.query(tables.ItemPocket).all():
|
||||
assert hit(filenames, media.ItemPocketMedia(root, item_pocket).icon)
|
||||
assert hit(filenames, media.ItemPocketMedia(root, item_pocket).icon, selected=True)
|
||||
assert hit(filenames, media.ItemPocketMedia(media_root, item_pocket).icon)
|
||||
assert hit(filenames, media.ItemPocketMedia(media_root, item_pocket).icon, selected=True)
|
||||
|
||||
for contest_type in session.query(tables.ContestType).all():
|
||||
assert hit(filenames, media.ContestTypeMedia(root, contest_type).icon)
|
||||
assert hit(filenames, media.ContestTypeMedia(media_root, contest_type).icon)
|
||||
|
||||
for elemental_type in session.query(tables.Type).all():
|
||||
assert hit(filenames, media.TypeMedia(root, elemental_type).icon)
|
||||
assert hit(filenames, media.TypeMedia(media_root, elemental_type).icon)
|
||||
|
||||
# Items
|
||||
versions_for_items = [
|
||||
|
|
@ -160,7 +146,7 @@ def test_get_everything(root, pytestconfig):
|
|||
]
|
||||
|
||||
for item in session.query(tables.Item).all():
|
||||
accessor = media.ItemMedia(root, item)
|
||||
accessor = media.ItemMedia(media_root, item)
|
||||
assert hit(filenames, accessor.berry_image) or not item.berry
|
||||
for rotation in (0, 90, 180, 270):
|
||||
assert hit(filenames, accessor.underground, rotation=rotation) or (
|
||||
|
|
@ -172,11 +158,11 @@ def test_get_everything(root, pytestconfig):
|
|||
|
||||
for color in 'red green blue pale prism'.split():
|
||||
for big in (True, False):
|
||||
accessor = media.UndergroundSphereMedia(root, color=color, big=big)
|
||||
accessor = media.UndergroundSphereMedia(media_root, color=color, big=big)
|
||||
assert hit(filenames, accessor.underground)
|
||||
|
||||
for rock_type in 'i ii o o-big s t z'.split():
|
||||
accessor = media.UndergroundRockMedia(root, rock_type)
|
||||
accessor = media.UndergroundRockMedia(media_root, rock_type)
|
||||
for rotation in (0, 90, 180, 270):
|
||||
success = hit(filenames, accessor.underground, rotation=rotation)
|
||||
assert success or rotation
|
||||
|
|
@ -184,17 +170,17 @@ def test_get_everything(root, pytestconfig):
|
|||
# Pokemon!
|
||||
accessors = []
|
||||
|
||||
accessors.append(media.UnknownPokemonMedia(root))
|
||||
accessors.append(media.EggMedia(root))
|
||||
accessors.append(media.UnknownPokemonMedia(media_root))
|
||||
accessors.append(media.EggMedia(media_root))
|
||||
manaphy = session.query(tables.PokemonSpecies).filter_by(identifier=u'manaphy').one()
|
||||
accessors.append(media.EggMedia(root, manaphy))
|
||||
accessors.append(media.SubstituteMedia(root))
|
||||
accessors.append(media.EggMedia(media_root, manaphy))
|
||||
accessors.append(media.SubstituteMedia(media_root))
|
||||
|
||||
for form in session.query(tables.PokemonForm).all():
|
||||
accessors.append(media.PokemonFormMedia(root, form))
|
||||
accessors.append(media.PokemonFormMedia(media_root, form))
|
||||
|
||||
for pokemon in session.query(tables.PokemonSpecies).all():
|
||||
accessors.append(media.PokemonSpeciesMedia(root, pokemon))
|
||||
accessors.append(media.PokemonSpeciesMedia(media_root, pokemon))
|
||||
|
||||
for accessor in accessors:
|
||||
assert hit(filenames, accessor.footprint) or not accessor.is_proper
|
||||
|
|
@ -243,9 +229,9 @@ def test_get_everything(root, pytestconfig):
|
|||
assert success
|
||||
|
||||
# Remove exceptions
|
||||
exceptions = [os.path.join(root, dirname) for dirname in
|
||||
exceptions = [os.path.join(media_root, dirname) for dirname in
|
||||
'chrome fonts ribbons'.split()]
|
||||
exceptions.append(os.path.join(root, 'items', 'hm-'))
|
||||
exceptions.append(os.path.join(media_root, 'items', 'hm-'))
|
||||
exceptions = tuple(exceptions)
|
||||
|
||||
unaccessed_filenames = set(filenames)
|
||||
|
|
|
|||
|
|
@ -1,9 +1,13 @@
|
|||
# encoding: utf8
|
||||
|
||||
import pokedex.roomaji
|
||||
from pokedex.tests import positional_params
|
||||
import pytest
|
||||
parametrize = pytest.mark.parametrize
|
||||
|
||||
@positional_params(
|
||||
import pokedex.roomaji
|
||||
|
||||
@parametrize(
|
||||
('kana', 'roomaji'),
|
||||
[
|
||||
(u'ヤミカラス', 'yamikarasu'),
|
||||
|
||||
# Elongated vowel
|
||||
|
|
@ -21,13 +25,16 @@ from pokedex.tests import positional_params
|
|||
(u'ラティアス', 'ratiasu'),
|
||||
(u'ウィー', 'wii'),
|
||||
(u'セレビィ', 'sereby'),
|
||||
]
|
||||
)
|
||||
def test_roomaji(kana, roomaji):
|
||||
result = pokedex.roomaji.romanize(kana)
|
||||
assert result == roomaji
|
||||
|
||||
|
||||
@positional_params(
|
||||
@parametrize(
|
||||
('kana', 'roomaji'),
|
||||
[
|
||||
(u'ヤミカラス', u'jamikarasu'),
|
||||
|
||||
# Elongated vowel
|
||||
|
|
@ -45,6 +52,7 @@ def test_roomaji(kana, roomaji):
|
|||
(u'ラティアス', u'ratiasu'),
|
||||
(u'ウィー', u'wí'),
|
||||
(u'セレビィ', u'serebí'),
|
||||
]
|
||||
)
|
||||
def test_roomaji_cs(kana, roomaji):
|
||||
result = pokedex.roomaji.romanize(kana, 'cs')
|
||||
|
|
|
|||
|
|
@ -1,6 +1,6 @@
|
|||
# encoding: utf8
|
||||
|
||||
from pokedex.tests import single_params
|
||||
import pytest
|
||||
|
||||
from sqlalchemy import Column, Integer, String, create_engine
|
||||
from sqlalchemy.orm import class_mapper, joinedload, sessionmaker
|
||||
|
|
@ -11,7 +11,9 @@ from pokedex.db import tables, markdown
|
|||
from pokedex.db.multilang import MultilangScopedSession, MultilangSession, \
|
||||
create_translation_table
|
||||
|
||||
@single_params(*dir(tables))
|
||||
parametrize = pytest.mark.parametrize
|
||||
|
||||
@parametrize('varname', [attr for attr in dir(tables) if not attr.startswith('_')])
|
||||
def test_variable_names(varname):
|
||||
"""We want pokedex.db.tables to export tables using the class name"""
|
||||
table = getattr(tables, varname)
|
||||
|
|
@ -24,7 +26,7 @@ def test_variable_names(varname):
|
|||
if classname and varname[0].isupper():
|
||||
assert varname == classname, '%s refers to %s' % (varname, classname)
|
||||
|
||||
@single_params(*tables.mapped_classes)
|
||||
@parametrize('table', tables.mapped_classes)
|
||||
def test_variable_names_2(table):
|
||||
"""We also want all of the tables exported"""
|
||||
assert getattr(tables, table.__name__) is table
|
||||
|
|
@ -162,7 +164,7 @@ classes = []
|
|||
for cls in tables.mapped_classes:
|
||||
classes.append(cls)
|
||||
classes += cls.translation_classes
|
||||
@single_params(*classes)
|
||||
@parametrize('cls', classes)
|
||||
def test_texts(cls):
|
||||
"""Check DB schema for integrity of text columns & translations.
|
||||
|
||||
|
|
@ -181,26 +183,27 @@ def test_texts(cls):
|
|||
format = column.info.get('format', None)
|
||||
if format is not None:
|
||||
if format not in good_formats:
|
||||
raise AssertionError(assert_text % column)
|
||||
pytest.fail(assert_text % column)
|
||||
if (format != 'identifier') and (column.name == 'identifier'):
|
||||
raise AssertionError('%s: identifier column name/type mismatch' % column)
|
||||
pytest.fail('%s: identifier column name/type mismatch' % column)
|
||||
if column.info.get('official', None) and format not in 'gametext plaintext':
|
||||
raise AssertionError('%s: official text with bad format' % column)
|
||||
pytest.fail('%s: official text with bad format' % column)
|
||||
text_columns.append(column)
|
||||
else:
|
||||
if isinstance(column.type, tables.Unicode):
|
||||
raise AssertionError('%s: text column without format' % column)
|
||||
pytest.fail('%s: text column without format' % column)
|
||||
if column.name == 'name' and format != 'plaintext':
|
||||
raise AssertionError('%s: non-plaintext name' % column)
|
||||
pytest.fail('%s: non-plaintext name' % column)
|
||||
# No mention of English in the description
|
||||
assert u'English' not in column.__doc__, column
|
||||
if column.doc and u'English' in column.doc:
|
||||
pytest.fail("%s: description mentions English" % column)
|
||||
# If there's more than one text column in a translation table,
|
||||
# they have to be nullable, to support missing translations
|
||||
if hasattr(cls, 'local_language') and len(text_columns) > 1:
|
||||
for column in text_columns:
|
||||
assert column.nullable
|
||||
|
||||
@single_params(*tables.mapped_classes)
|
||||
@parametrize('table', tables.mapped_classes)
|
||||
def test_identifiers_with_names(table):
|
||||
"""Test that named tables have identifiers
|
||||
"""
|
||||
|
|
|
|||
|
|
@ -1,21 +1,24 @@
|
|||
# Encoding: UTF-8
|
||||
|
||||
import pytest
|
||||
from sqlalchemy.orm.exc import NoResultFound
|
||||
parametrize = pytest.mark.parametrize
|
||||
|
||||
from pokedex.tests import positional_params
|
||||
from sqlalchemy.orm.exc import NoResultFound
|
||||
|
||||
from pokedex.db import tables, connect, util, markdown
|
||||
|
||||
connection = connect()
|
||||
@pytest.fixture(scope="module")
|
||||
def session(request):
|
||||
uri = request.config.getvalue("engine")
|
||||
return connect(uri)
|
||||
|
||||
def test_filter():
|
||||
q = connection.query(tables.PokemonSpecies).filter(
|
||||
def test_filter(session):
|
||||
q = session.query(tables.PokemonSpecies).filter(
|
||||
tables.PokemonSpecies.name == u"Marowak")
|
||||
assert q.one().identifier == 'marowak'
|
||||
|
||||
def test_languages():
|
||||
q = connection.query(tables.PokemonSpecies).filter(
|
||||
def test_languages(session):
|
||||
q = session.query(tables.PokemonSpecies).filter(
|
||||
tables.PokemonSpecies.name == u"Mightyena")
|
||||
pkmn = q.one()
|
||||
for lang, name in (
|
||||
|
|
@ -24,35 +27,35 @@ def test_languages():
|
|||
('roomaji', u'Guraena'),
|
||||
('fr', u'Grahyèna'),
|
||||
):
|
||||
language = connection.query(tables.Language).filter_by(
|
||||
language = session.query(tables.Language).filter_by(
|
||||
identifier=lang).one()
|
||||
assert pkmn.name_map[language] == name
|
||||
|
||||
def test_bad_lang():
|
||||
def test_bad_lang(session):
|
||||
with pytest.raises(KeyError):
|
||||
q = connection.query(tables.PokemonSpecies).filter(
|
||||
q = session.query(tables.PokemonSpecies).filter(
|
||||
tables.PokemonSpecies.name == u"Mightyena")
|
||||
pkmn = q.one()
|
||||
pkmn.names["identifier of a language that doesn't exist"]
|
||||
|
||||
def test_mutating():
|
||||
item = connection.query(tables.Item).filter_by(
|
||||
def test_mutating(session):
|
||||
item = session.query(tables.Item).filter_by(
|
||||
identifier=u"jade-orb").one()
|
||||
language = connection.query(tables.Language).filter_by(
|
||||
language = session.query(tables.Language).filter_by(
|
||||
identifier=u"de").one()
|
||||
item.name_map[language] = u"foo"
|
||||
assert item.name_map[language] == "foo"
|
||||
item.name_map[language] = u"xyzzy"
|
||||
assert item.name_map[language] == "xyzzy"
|
||||
|
||||
def test_mutating_default():
|
||||
item = connection.query(tables.Item).filter_by(
|
||||
def test_mutating_default(session):
|
||||
item = session.query(tables.Item).filter_by(
|
||||
identifier=u"jade-orb").one()
|
||||
item.name = u"foo"
|
||||
assert item.name == "foo"
|
||||
|
||||
def test_string_mapping():
|
||||
item = connection.query(tables.Item).filter_by(
|
||||
def test_string_mapping(session):
|
||||
item = session.query(tables.Item).filter_by(
|
||||
identifier=u"jade-orb").one()
|
||||
assert len(item.name_map) == len(item.names)
|
||||
for lang in item.names:
|
||||
|
|
@ -61,25 +64,25 @@ def test_string_mapping():
|
|||
assert "language that doesn't exist" not in item.name_map
|
||||
assert tables.Language() not in item.name_map
|
||||
|
||||
def test_new_language():
|
||||
item = connection.query(tables.Item).filter_by(
|
||||
def test_new_language(session):
|
||||
item = session.query(tables.Item).filter_by(
|
||||
identifier=u"jade-orb").one()
|
||||
language = tables.Language()
|
||||
language.id = -1
|
||||
language.identifier = u'test'
|
||||
language.iso639 = language.iso3166 = u'--'
|
||||
language.official = False
|
||||
connection.add(language)
|
||||
session.add(language)
|
||||
item.name_map[language] = u"foo"
|
||||
assert item.name_map[language] == "foo"
|
||||
assert language in item.name_map
|
||||
item.name_map[language] = u"xyzzy"
|
||||
assert item.name_map[language] == "xyzzy"
|
||||
|
||||
def test_markdown():
|
||||
move = connection.query(tables.Move).filter_by(
|
||||
def test_markdown(session):
|
||||
move = session.query(tables.Move).filter_by(
|
||||
identifier=u"thunderbolt").one()
|
||||
language = connection.query(tables.Language).filter_by(
|
||||
language = session.query(tables.Language).filter_by(
|
||||
identifier=u"en").one()
|
||||
assert '10%' in move.effect.as_text()
|
||||
assert '10%' in move.effect_map[language].as_text()
|
||||
|
|
@ -90,9 +93,9 @@ def test_markdown():
|
|||
assert '10%' in move.effect.__html__()
|
||||
assert '10%' in move.effect_map[language].__html__()
|
||||
|
||||
def test_markdown_string():
|
||||
en = util.get(connection, tables.Language, 'en')
|
||||
md = markdown.MarkdownString('[]{move:thunderbolt} [paralyzes]{mechanic:paralysis} []{form:sky shaymin}. []{pokemon:mewthree} does not exist.', connection, en)
|
||||
def test_markdown_string(session):
|
||||
en = util.get(session, tables.Language, 'en')
|
||||
md = markdown.MarkdownString('[]{move:thunderbolt} [paralyzes]{mechanic:paralysis} []{form:sky shaymin}. []{pokemon:mewthree} does not exist.', session, en)
|
||||
assert unicode(md) == 'Thunderbolt paralyzes Sky Shaymin. mewthree does not exist.'
|
||||
assert md.as_html() == '<p><span>Thunderbolt</span> <span>paralyzes</span> <span>Sky Shaymin</span>. <span>mewthree</span> does not exist.</p>'
|
||||
|
||||
|
|
@ -108,9 +111,9 @@ def test_markdown_string():
|
|||
def identifier_url(self, category, ident):
|
||||
return "%s/%s" % (category, ident)
|
||||
|
||||
assert md.as_html(extension=ObjectTestExtension(connection)) == (
|
||||
assert md.as_html(extension=ObjectTestExtension(session)) == (
|
||||
'<p><a href="move/thunderbolt">Thunderbolt</a> <span>paralyzes</span> <a href="form/sky shaymin">Sky Shaymin</a>. <span>mewthree</span> does not exist.</p>')
|
||||
assert md.as_html(extension=IdentifierTestExtension(connection)) == (
|
||||
assert md.as_html(extension=IdentifierTestExtension(session)) == (
|
||||
'<p><a href="move/thunderbolt">Thunderbolt</a> <a href="mechanic/paralysis">paralyzes</a> <a href="form/sky shaymin">Sky Shaymin</a>. <a href="pokemon/mewthree">mewthree</a> does not exist.</p>')
|
||||
|
||||
def markdown_column_params():
|
||||
|
|
@ -130,10 +133,14 @@ def markdown_column_params():
|
|||
if column.info.get('string_getter') == markdown.MarkdownString:
|
||||
yield cls, translation_cls, column.name
|
||||
|
||||
@positional_params(*markdown_column_params())
|
||||
def test_markdown_values(parent_class, translation_class, column_name):
|
||||
@pytest.mark.slow
|
||||
@parametrize(
|
||||
('parent_class', 'translation_class', 'column_name'),
|
||||
list(markdown_column_params())
|
||||
)
|
||||
def test_markdown_values(session, parent_class, translation_class, column_name):
|
||||
"""Implementation for the above"""
|
||||
query = connection.query(parent_class)
|
||||
query = session.query(parent_class)
|
||||
if translation_class:
|
||||
query = query.join(translation_class)
|
||||
|
||||
|
|
@ -150,7 +157,7 @@ def test_markdown_values(parent_class, translation_class, column_name):
|
|||
'%s: unknown link target: {%s:%s}' %
|
||||
(key, category, ident))
|
||||
|
||||
test_extension = TestExtension(connection)
|
||||
test_extension = TestExtension(session)
|
||||
|
||||
for item in query:
|
||||
for language, md_text in getattr(item, column_name + '_map').items():
|
||||
|
|
|
|||
|
|
@ -2,13 +2,14 @@
|
|||
|
||||
import csv
|
||||
|
||||
import pytest
|
||||
|
||||
from pokedex.db import translations, tables
|
||||
|
||||
fake_version_names = (
|
||||
'version_id,local_language_id,name',
|
||||
'1,0,name1', '2,0,name2', '3,0,name3', '3,1,othername3',
|
||||
'1,0,name1',
|
||||
'2,0,name2',
|
||||
'3,0,name3',
|
||||
'3,1,othername3',
|
||||
)
|
||||
|
||||
fake_translation_csv = (
|
||||
|
|
|
|||
|
|
@ -1,49 +1,47 @@
|
|||
# Encoding: utf8
|
||||
|
||||
import pytest
|
||||
parametrize = pytest.mark.parametrize
|
||||
|
||||
from pokedex.tests import single_params
|
||||
from pokedex.db import connect, tables, util
|
||||
|
||||
session = connect()
|
||||
|
||||
def test_get_item_identifier():
|
||||
def test_get_item_identifier(session):
|
||||
item = util.get(session, tables.Item, identifier='master-ball')
|
||||
assert item.name == 'Master Ball'
|
||||
|
||||
def test_get_item_name():
|
||||
def test_get_item_name(session):
|
||||
item = util.get(session, tables.Item, name='Awakening')
|
||||
assert item.name == 'Awakening'
|
||||
|
||||
def test_get_english_by_identifier():
|
||||
def test_get_english_by_identifier(session):
|
||||
language = util.get(session, tables.Language, 'en')
|
||||
assert language.name == 'English'
|
||||
|
||||
@single_params(*'burmy shaymin unown cresselia'.split())
|
||||
def test_get_pokemon_identifier(identifier):
|
||||
@parametrize('identifier', ['burmy', 'shaymin', 'unown', 'cresselia'])
|
||||
def test_get_pokemon_identifier(session, identifier):
|
||||
poke = util.get(session, tables.PokemonSpecies, identifier=identifier)
|
||||
assert poke.identifier == identifier
|
||||
|
||||
@single_params(*'Burmy Shaymin Unown Cresselia'.split())
|
||||
def test_get_pokemon_name(name):
|
||||
@parametrize('name', ['Burmy', 'Shaymin', 'Unown', 'Cresselia'])
|
||||
def test_get_pokemon_name(session, name):
|
||||
poke = util.get(session, tables.PokemonSpecies, name=name)
|
||||
assert poke.name == name
|
||||
|
||||
@single_params(*'Cheniti Shaymin Zarbi Cresselia'.split())
|
||||
def test_get_pokemon_name_explicit_language(name):
|
||||
@parametrize('name', ['Cheniti', 'Shaymin', 'Zarbi', 'Cresselia'])
|
||||
def test_get_pokemon_name_explicit_language(session, name):
|
||||
french = util.get(session, tables.Language, 'fr')
|
||||
poke = util.get(session, tables.PokemonSpecies, name=name, language=french)
|
||||
assert poke.name_map[french] == name, poke.name_map[french]
|
||||
|
||||
def test_types_french_order():
|
||||
def test_types_french_order(session):
|
||||
french = util.get(session, tables.Language, 'fr')
|
||||
types = session.query(tables.Type).filter(tables.Type.id < 10000)
|
||||
types = list(util.order_by_name(types, tables.Type, language=french))
|
||||
assert types[0].name_map[french] == 'Acier', types[0].name_map[french]
|
||||
assert types[-1].name_map[french] == 'Vol', types[-1].name_map[french]
|
||||
|
||||
@single_params(*range(1, 10) * 2)
|
||||
def test_get_pokemon_id(id):
|
||||
@parametrize('id', range(1, 10))
|
||||
def test_get_pokemon_id(session, id):
|
||||
result = util.get(session, tables.Pokemon, id=id)
|
||||
assert result.id == id
|
||||
assert result.__tablename__ == 'pokemon'
|
||||
|
|
|
|||
Loading…
Reference in a new issue