Repaired tests to correctly run under Python 3.3

miguelgrinberg · Apr 4, 2014 · d4451fd · d4451fd
1 parent 87192f1
commit d4451fd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tests/test_migrate.py b/tests/test_migrate.py
@@ -1,6 +1,5 @@
 import os
 import shutil
-from exceptions import OSError
 import unittest
 
 class TestMigrate(unittest.TestCase):
@@ -30,7 +29,7 @@ def tearDown(self):
             pass
 
     def test_migrate_upgrade(self):
-        from app import db, User
+        from .app import db, User
         db.session.add(User(name = 'test'))
         db.session.commit()
 

diff --git a/tests/test_migrate_custom_directory.py b/tests/test_migrate_custom_directory.py
@@ -1,6 +1,5 @@
 import os
 import shutil
-from exceptions import OSError
 import unittest
 
 class TestMigrate(unittest.TestCase):
@@ -30,7 +29,7 @@ def tearDown(self):
             pass
 
     def test_migrate_upgrade(self):
-        from app2 import db, User
+        from .app2 import db, User
         db.session.add(User(name = 'test'))
         db.session.commit()