mirror of
https://github.com/veekun/pokedex.git
synced 2024-08-20 18:16:34 +00:00
Added random lookup. #57
This commit is contained in:
parent
9e3d8b317d
commit
f5853c0a0b
2 changed files with 71 additions and 11 deletions
pokedex/tests
|
@ -102,3 +102,14 @@ def test_wildcard_lookup():
|
|||
first_result = results[0]
|
||||
assert_equal(first_result.object.name, name,
|
||||
u'Wildcards work correctly')
|
||||
|
||||
def test_random_lookup():
|
||||
for _ in xrange(5):
|
||||
results = pokedex.lookup.lookup(u'random')
|
||||
assert_equal(len(results), 1, u'Random returns one result')
|
||||
|
||||
for table_name in [u'pokemon', u'moves', u'items', u'abilities', u'types']:
|
||||
results = pokedex.lookup.lookup(u'random', valid_types=[table_name])
|
||||
assert_equal(len(results), 1, u'Constrained random returns one result')
|
||||
assert_equal(results[0].object.__tablename__, table_name,
|
||||
u'Constrained random returns result from the right table')
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue