From bd70b19c7408e29a74a15e1288db1821f58cf777 Mon Sep 17 00:00:00 2001 From: Eevee Date: Sun, 23 Aug 2009 20:43:39 -0700 Subject: [PATCH] Added lookup tests. #15 --- pokedex/tests/__init__.py | 15 ++++++++ pokedex/tests/test_lookup.py | 91 ++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 106 insertions(+) create mode 100644 pokedex/tests/__init__.py create mode 100644 pokedex/tests/test_lookup.py diff --git a/pokedex/tests/__init__.py b/pokedex/tests/__init__.py new file mode 100644 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 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"') -- 2.7.4