Separate the tests that hit the database from the rest of the tests.

[dead/census-tools.git] / src / Tests / Unit / CensusTest.py

diff --git a/src/Tests/Unit/CensusTest.py b/src/Tests/Unit/CensusTest.py
index 0789e2fe6b897357f359399aeb5cff2f32402f71..531d4dd0a615a2d7d27afbee8f0e36c5ebbe2cf1 100644 (file)
--- a/src/Tests/Unit/CensusTest.py
+++ b/src/Tests/Unit/CensusTest.py
@@ -32,7 +32,13 @@ class DatabaseTest(unittest.TestCase):
             
     
 def suite():
+    # Tests that don't hit the database.
     suite = unittest.TestSuite()
-    suite.addTest(unittest.makeSuite(DatabaseTest))
     return suite
-        
+
+
+def db_suite():
+    # The tests that hit the database.
+    db_suite = unittest.TestSuite()
+    db_suite.addTest(unittest.makeSuite(DatabaseTest))
+    return db_suite