Added lookup tests. #15
[zzz-pokedex.git] / pokedex / tests / test_lookup.py
1 # encoding: utf8
2 from nose.tools import *
3 import unittest
4
5 import pokedex.lookup
6
7
8 def test_exact_lookup():
9 tests = [
10 # Simple lookups
11 (u'Eevee', 'pokemon', 133),
12 (u'Scratch', 'moves', 10),
13 (u'Master Ball', 'items', 1),
14 (u'normal', 'types', 1),
15 (u'Run Away', 'abilities', 50),
16
17 # Funny characters
18 (u'Mr. Mime', 'pokemon', 122),
19 (u"Farfetch'd", 'pokemon', 83),
20 (u'Poké Ball', 'items', 4),
21
22 # Forms
23 (u'Rotom', 'pokemon', 479),
24 (u'Wash Rotom', 'pokemon', 504),
25
26 # Other languages
27 (u'イーブイ', 'pokemon', 133),
28 (u'Iibui', 'pokemon', 133),
29 (u'Eievui', 'pokemon', 133),
30 (u'이브이', 'pokemon', 133),
31 (u'伊布', 'pokemon', 133),
32 (u'Evoli', 'pokemon', 133),
33 ]
34
35 for input, table, id in tests:
36 results = pokedex.lookup.lookup(input)
37 assert_equal(len(results), 1, u"'%s' returns one result" % input)
38 assert_equal(results[0].exact, True, u"'%s' match exactly" % input)
39
40 row = results[0].object
41 assert_equal(row.__tablename__, table, u"'%s' is in the right table" % input)
42 assert_equal(row.id, id, u"'%s' returns the right id" % input)
43
44
45 def test_id_lookup():
46 results = pokedex.lookup.lookup(u'1')
47 assert_true(len(results) >= 5, u'At least five things have id 1')
48 assert_true(all(_.object.id == 1 for _ in results),
49 u'All results have id 1')
50
51 def test_multi_lookup():
52 results = pokedex.lookup.lookup(u'Metronome')
53 assert_equal(len(results), 2, u'Two things called "Metronome"')
54 assert_true(results[0].exact, u'Metronome matches are exact')
55
56
57 def test_type_lookup():
58 results = pokedex.lookup.lookup(u'pokemon:1')
59 assert_equal(results[0].object.__tablename__, 'pokemon',
60 u'Type restriction works correctly')
61 assert_equal(len(results), 1, u'Only one id result when type is specified')
62 assert_equal(results[0].name, u'Bulbasaur', u'Type + id returns the right result')
63
64 results = pokedex.lookup.lookup(u'1', valid_types=['pokemon'])
65 assert_equal(results[0].name, u'Bulbasaur', u'valid_types works as well as type: prefix')
66
67 def test_fuzzy_lookup():
68 tests = [
69 # Regular English names
70 (u'chamander', u'Charmander'),
71 (u'pokeball', u'Poké Ball'),
72
73 # Names with squiggles in them
74 (u'farfetchd', u"Farfetch'd"),
75 (u'porygonz', u'Porygon-Z'),
76
77 # Sufficiently long foreign names
78 (u'カクレオ', u'Kecleon'),
79 (u'Yamikrasu', u'Murkrow'),
80 ]
81
82 for misspelling, name in tests:
83 results = pokedex.lookup.lookup(misspelling)
84 first_result = results[0]
85 assert_equal(first_result.object.name, name,
86 u'Simple misspellings are corrected')
87
88 results = pokedex.lookup.lookup(u'Nidoran')
89 top_names = [_.object.name for _ in results[0:2]]
90 assert_true(u'Nidoran♂' in top_names, u'Nidoran♂ is a top result for "Nidoran"')
91 assert_true(u'Nidoran♀' in top_names, u'Nidoran♀ is a top result for "Nidoran"')