fix tests.

diegorubin · Dec 10, 2015 · 089acef · 089acef
1 parent 5309410
commit 089acef
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/tests/test_spellchecker.py b/tests/test_spellchecker.py
@@ -16,8 +16,10 @@
 
 class TestSpellchecker(unittest.TestCase):
     def setUp(self):
+        word = Word('abacaxi', 'SYSTEM-TEST')
+        word.save()
+
         self.spellchecker = Spellchecker('pt_BR.list')
-        word = Word('abacaxi', 'TEST')
 
     def test_correct(self):
         self.assertEqual('abacaxi', self.spellchecker.correct('abacahi'))

diff --git a/tests/test_word_list.py b/tests/test_word_list.py
@@ -14,6 +14,10 @@
     from spellchecker import word_list
 
 class TestWord(unittest.TestCase):
+    def setUp(self):
+        word = Word('cachorro', 'SYSTEM-TEST')
+        word.save()
+
     def test_save_word(self):
         word = Word('cachorro', 'SYSTEM-TEST')
         self.assertEqual(True, word.save())