X-Git-Url: http://git.veekun.com/zzz-pokedex.git/blobdiff_plain/c8551bf2e5b4f33200ca91b6e8c4cf4a5fc109ee..d131ce6a550d4d3572584c04984550d151eccd7b:/pokedex/tests/test_lookup.py

diff --git a/pokedex/tests/test_lookup.py b/pokedex/tests/test_lookup.py
index 6479bd6..094e630 100644
--- a/pokedex/tests/test_lookup.py
+++ b/pokedex/tests/test_lookup.py
@@ -2,8 +2,15 @@
 from nose.tools import *
 import unittest
 
-import pokedex.lookup
+from pokedex.lookup import PokedexLookup
 
+lookup = None
+
+def setup():
+    # Recreate data
+    global lookup
+    lookup = PokedexLookup()
+    lookup.rebuild_index()
 
 def test_exact_lookup():
     tests = [
@@ -33,7 +40,7 @@ def test_exact_lookup():
     ]
 
     for input, table, id in tests:
-        results = pokedex.lookup.lookup(input)
+        results = lookup.lookup(input)
         assert_equal(len(results), 1,           u"'%s' returns one result" % input)
         assert_equal(results[0].exact, True,    u"'%s' match exactly" % input)
 
@@ -43,27 +50,47 @@ def test_exact_lookup():
 
 
 def test_id_lookup():
-    results = pokedex.lookup.lookup(u'1')
+    results = lookup.lookup(u'1')
     assert_true(len(results) >= 5,              u'At least five things have id 1')
     assert_true(all(_.object.id == 1 for _ in results),
                                                 u'All results have id 1')
 
 def test_multi_lookup():
-    results = pokedex.lookup.lookup(u'Metronome')
+    results = lookup.lookup(u'Metronome')
     assert_equal(len(results), 2,               u'Two things called "Metronome"')
     assert_true(results[0].exact,               u'Metronome matches are exact')
 
 
 def test_type_lookup():
-    results = pokedex.lookup.lookup(u'pokemon:1')
+    results = lookup.lookup(u'pokemon:1')
     assert_equal(results[0].object.__tablename__, 'pokemon',
                                                 u'Type restriction works correctly')
     assert_equal(len(results), 1,               u'Only one id result when type is specified')
     assert_equal(results[0].name, u'Bulbasaur', u'Type + id returns the right result')
 
-    results = pokedex.lookup.lookup(u'1', valid_types=['pokemon'])
+    results = lookup.lookup(u'1', valid_types=['pokemon'])
     assert_equal(results[0].name, u'Bulbasaur', u'valid_types works as well as type: prefix')
 
+def test_language_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')
+    assert_true(len(results) > 1,               u'There are multiple "charge"s')
+
+    results = lookup.lookup(u'@fr:charge')
+    assert_equal(results[0].iso639, u'fr',      u'Language restriction works correctly')
+    assert_equal(len(results), 1,               u'Only one "charge" result when language is specified')
+    assert_equal(results[0].object.name, u'Tackle',
+                                                u'Language + vague name returns the right result')
+
+    results = lookup.lookup(u'charge', valid_types=['@fr'])
+    assert_equal(results[0].object.name, u'Tackle',
+                                                u'valid_types works as well as @lang: prefix')
+
+    results = lookup.lookup(u'@fr,move:charge')
+    assert_equal(results[0].object.name, u'Tackle',
+                                                u'Languages and types both work together')
+
 def test_fuzzy_lookup():
     tests = [
         # Regular English names
@@ -80,12 +107,12 @@ def test_fuzzy_lookup():
     ]
 
     for misspelling, name in tests:
-        results = pokedex.lookup.lookup(misspelling)
+        results = lookup.lookup(misspelling)
         first_result = results[0]
         assert_equal(first_result.object.name, name,
                                                 u'Simple misspellings are corrected')
 
-    results = pokedex.lookup.lookup(u'Nidoran')
+    results = lookup.lookup(u'Nidoran')
     top_names = [_.object.name for _ in results[0:2]]
     assert_true(u'Nidoran♂' in top_names,       u'Nidoran♂ is a top result for "Nidoran"')
     assert_true(u'Nidoran♀' in top_names,       u'Nidoran♀ is a top result for "Nidoran"')
@@ -98,18 +125,18 @@ def test_wildcard_lookup():
     ]
 
     for wildcard, name in tests:
-        results = pokedex.lookup.lookup(wildcard)
+        results = lookup.lookup(wildcard)
         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')
+        results = 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])
+        results = 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')