jbparse/test/kanjidic2.py

   1 # -*- coding: utf-8 -*-
   2
   3 from __future__ import absolute_import
   4
   5 import unittest, time
   6 from jbparse import kanjidic2
   7
   8 SRC_NAME = "kanjidic2.xml"
   9 SRC_DIR = "../../dicts"
  10
  11 SRC_NAME = "/".join((SRC_DIR, SRC_NAME))
  12
  13 class Kanjidic2Test(unittest.TestCase):
  14
  15     def setUp(self):
  16         self.parser = kanjidic2.Parser(SRC_NAME)
  17
  18     def test_single_kanji_search(self):
  19         """KANJIDIC2: Search for single kanji"""
  20         query = u"食"
  21         l = [entry for entry in self.parser.search(query)]
  22         self.assertEqual(len(l), 1)
  23         self.assertEqual(query, l[0].literal)
  24
  25     def test_multi_kanji_search(self):
  26         """KANJIDIC2: Search for multiple kanji at once"""
  27         query = u"上下"
  28         l = [entry for entry in self.parser.search(query)]
  29         self.assertEqual(len(l), 2)
  30         for char in query:
  31             self.assertTrue(char in [entry.literal for entry in l])
  32
  33     def test_mixed_search(self):
  34         """KANJIDIC2: Search query with kanji and non-kanji characters"""
  35         query = u"天気はいいから、散歩しましょう。  Right?"
  36         l = [entry for entry in self.parser.search(query)]
  37         self.assertEqual(len(l), 4)
  38         for char in u"天気散歩":
  39             self.assertTrue(char in [entry.literal for entry in l])
  40
  41     def tearDown(self):
  42         self.parser = None
  43
  44 if __name__ == "__main__":
  45     unittest.main()