test/spellcheck_test.py

   1 #!/usr/bin/env python
   2 from __future__ import absolute_import, division, unicode_literals
   3 import unittest
   4
   5 from cola import compat
   6 from cola import spellcheck
   7
   8 from . import helper
   9
  10
  11 class TestCase(unittest.TestCase):
  12
  13     def test_spellcheck_generator(self):
  14         check = spellcheck.NorvigSpellCheck()
  15         self.assert_spellcheck(check)
  16
  17     def test_spellcheck_unicode(self):
  18         path = helper.fixture('unicode.txt')
  19         check = spellcheck.NorvigSpellCheck(cracklib=path)
  20         self.assert_spellcheck(check)
  21
  22     def assert_spellcheck(self, check):
  23         for word in check.read():
  24             self.assertTrue(word is not None)
  25             self.assertTrue(isinstance(word, compat.ustr))
  26
  27
  28 if __name__ == '__main__':
  29     unittest.main()