From 1d8928e4a0d126644e538fb18c1ed243d298bf8a Mon Sep 17 00:00:00 2001
From: Friedrich Lindenberg <friedrich@pudo.org>
Date: Mon, 8 Jun 2015 10:04:30 +0200
Subject: [PATCH] Add facility to run tests directly, fixes #129.

---
 test/test_freeze.py        | 6 +++++-
 test/test_freeze_app.py    | 6 +++++-
 test/test_freeze_config.py | 4 ++++
 test/test_persistence.py   | 6 +++++-
 4 files changed, 19 insertions(+), 3 deletions(-)

diff --git a/test/test_freeze.py b/test/test_freeze.py
index 9605a29..16f640e 100644
--- a/test/test_freeze.py
+++ b/test/test_freeze.py
@@ -12,7 +12,7 @@ from dataset import connect
 from dataset.freeze.app import freeze
 from dataset.freeze.format.fcsv import value_to_str
 
-from .sample_data import TEST_DATA
+from sample_data import TEST_DATA
 
 
 class FreezeTestCase(unittest.TestCase):
@@ -94,3 +94,7 @@ class SerializerTestCase(unittest.TestCase):
 
     def test_value_to_str4(self):
         assert [] == value_to_str([])
+
+
+if __name__ == '__main__':
+    unittest.main()
diff --git a/test/test_freeze_app.py b/test/test_freeze_app.py
index 20826f6..a559895 100644
--- a/test/test_freeze_app.py
+++ b/test/test_freeze_app.py
@@ -14,7 +14,7 @@ from dataset import connect
 from dataset.util import FreezeException
 from dataset.freeze.config import Configuration, Export
 from dataset.freeze.app import create_parser, freeze_with_config, freeze_export
-from .sample_data import TEST_DATA
+from sample_data import TEST_DATA
 
 
 class FreezeAppTestCase(TestCase):
@@ -61,3 +61,7 @@ class FreezeAppTestCase(TestCase):
         cfg.data['common']['query'] = 'SELECT * FROM weather'
         cfg.data['exports'] = [{'filename': 'weather.csv', 'format': 'csv'}]
         freeze_with_config(cfg, db=self.db)
+
+
+if __name__ == '__main__':
+    unittest.main()
diff --git a/test/test_freeze_config.py b/test/test_freeze_config.py
index f892c03..67d996e 100644
--- a/test/test_freeze_config.py
+++ b/test/test_freeze_config.py
@@ -33,3 +33,7 @@ class TestConfiguration(unittest.TestCase):
         self.assertRaises(FreezeException, list, cfg.exports)
         cfg.data = {}
         self.assertRaises(FreezeException, list, cfg.exports)
+
+
+if __name__ == '__main__':
+    unittest.main()
diff --git a/test/test_persistence.py b/test/test_persistence.py
index 6c8cf4c..92baf2f 100644
--- a/test/test_persistence.py
+++ b/test/test_persistence.py
@@ -12,7 +12,7 @@ from sqlalchemy.exc import IntegrityError, SQLAlchemyError
 from dataset import connect
 from dataset.util import DatasetException
 
-from .sample_data import TEST_DATA, TEST_CITY_1
+from sample_data import TEST_DATA, TEST_CITY_1
 
 
 class DatabaseTestCase(unittest.TestCase):
@@ -404,3 +404,7 @@ class RowTypeTestCase(unittest.TestCase):
             c += 1
             assert isinstance(row, Constructor), row
         assert c == len(self.tbl)
+
+
+if __name__ == '__main__':
+    unittest.main()