Added lookup tests. #15

diff --git a/pokedex/tests/__init__.py b/pokedex/tests/__init__.py
new file mode 100644 (file)
index 0000000..29b7a56
--- /dev/null
+++ b/pokedex/tests/__init__.py
@@ -0,0 +1,15 @@
+import unittest
+
+from pokedex.lookup import open_index
+from pokedex.db import connect
+from pokedex.db.load import load
+
+def setup():
+    # Reload data just in case
+    session = connect()
+    load(session, verbose=False, drop_tables=True)
+    open_index(session=session, recreate=True)
+
+
+def teardown():
+    print "teardown"

diff --git a/pokedex/tests/test_lookup.py b/pokedex/tests/test_lookup.py
new file mode 100644 (file)
index 0000000..b6bef60
--- /dev/null
+++ b/pokedex/tests/test_lookup.py
@@ -0,0 +1,91 @@
+# encoding: utf8
+from nose.tools import *
+import unittest
+
+import pokedex.lookup
+
+
+def test_exact_lookup():
+    tests = [
+        # Simple lookups
+        (u'Eevee',          'pokemon',      133),
+        (u'Scratch',        'moves',        10),
+        (u'Master Ball',    'items',        1),
+        (u'normal',         'types',        1),
+        (u'Run Away',       'abilities',    50),
+
+        # Funny characters
+        (u'Mr. Mime',       'pokemon',      122),
+        (u"Farfetch'd",     'pokemon',      83),
+        (u'Poké Ball',      'items',        4),
+
+        # Forms
+        (u'Rotom',          'pokemon',      479),
+        (u'Wash Rotom',     'pokemon',      504),
+
+        # Other languages
+        (u'イーブイ',       'pokemon',      133),
+        (u'Iibui',          'pokemon',      133),
+        (u'Eievui',         'pokemon',      133),
+        (u'이브이',         'pokemon',      133),
+        (u'伊布',           'pokemon',      133),
+        (u'Evoli',          'pokemon',      133),
+    ]
+
+    for input, table, id in tests:
+        results = pokedex.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)
+
+        row = results[0].object
+        assert_equal(row.__tablename__, table,  u"'%s' is in the right table" % input)
+        assert_equal(row.id, id,                u"'%s' returns the right id" % input)
+
+
+def test_id_lookup():
+    results = pokedex.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')
+    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')
+    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'])
+    assert_equal(results[0].name, u'Bulbasaur', u'valid_types works as well as type: prefix')
+
+def test_fuzzy_lookup():
+    tests = [
+        # Regular English names
+        (u'chamander',          u'Charmander'),
+        (u'pokeball',           u'Poké Ball'),
+
+        # Names with squiggles in them
+        (u'farfetchd',          u"Farfetch'd"),
+        (u'porygonz',           u'Porygon-Z'),
+
+        # Sufficiently long foreign names
+        (u'カクレオ',           u'Kecleon'),
+        (u'Yamikrasu',          u'Murkrow'),
+    ]
+
+    for misspelling, name in tests:
+        results = pokedex.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')
+    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"')