@@ -62,21 +62,19 @@ class TestLanguageTool(unittest.TestCase):
6262 },
6363 }
6464
65- def setUp (self ):
66- self .lang_check = language_check .LanguageTool ()
67-
6865 def test_check (self ):
66+ lang_check = language_check .LanguageTool ()
6967 for language , tests in self .check_tests .items ():
7068 try :
71- self . lang_check .language = language
69+ lang_check .language = language
7270 except ValueError :
7371 version = language_check .get_version ()
7472 warnings .warn (
7573 'LanguageTool {} doesn’t support language {!r}'
7674 .format (version , language )
7775 )
7876 for text , expected_matches in tests :
79- matches = self . lang_check .check (text )
77+ matches = lang_check .check (text )
8078 for expected_match in expected_matches :
8179 for match in matches :
8280 if (
@@ -90,17 +88,18 @@ def test_check(self):
9088 'can’t find {!r}' .format (expected_match ))
9189
9290 def test_correct (self ):
91+ lang_check = language_check .LanguageTool ()
9392 for language , tests in self .correct_tests .items ():
9493 try :
95- self . lang_check .language = language
94+ lang_check .language = language
9695 except ValueError :
9796 version = language_check .get_version ()
9897 warnings .warn (
9998 'LanguageTool {} doesn’t support language {!r}'
10099 .format (version , language )
101100 )
102101 for text , result in tests .items ():
103- self .assertEqual (self . lang_check .correct (text ), result )
102+ self .assertEqual (lang_check .correct (text ), result )
104103
105104 def test_languages (self ):
106105 self .assertIn ('en' , language_check .get_languages ())
@@ -116,6 +115,16 @@ def test_get_directory(self):
116115 language_check .set_directory (path )
117116 self .assertEqual (path , language_check .get_directory ())
118117
118+ def test_disable_spellcheck (self ):
119+ lang_check = language_check .LanguageTool ()
120+ self .assertTrue (lang_check .check ('helo' ))
121+
122+ lang_check .disable_spellchecking ()
123+ self .assertFalse (lang_check .check ('helo' ))
124+
125+ lang_check .enable_spellchecking ()
126+ self .assertTrue (lang_check .check ('helo' ))
127+
119128
120129if __name__ == '__main__' :
121130 unittest .main ()
0 commit comments