add __main__ section

mozilla · Jul 20, 2012 · 689c19a · 689c19a
1 parent 1befa48
commit 689c19a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/dzclient/tests/test_datazilla_request.py b/dzclient/tests/test_datazilla_request.py
@@ -1,12 +1,9 @@
-from unittest import TestCase
+import unittest
-
 from mock import patch
-
 from dzclient import DatazillaRequest, DatazillaResult
 
 
-
+class DatazillaRequestTest(unittest.TestCase):
-class DatazillaRequestTest(TestCase):
     def test_init_with_date(self):
         """Can provide test date on instantiation."""
         req = DatazillaRequest(
@@ -144,3 +141,6 @@ def test_send(self,
             header['Content-type'],
             'application/x-www-form-urlencoded',
             )
+
+if __name__ == '__main__':
+    unittest.main()
diff --git a/dzclient/tests/test_datazilla_result.py b/dzclient/tests/test_datazilla_result.py
@@ -1,10 +1,8 @@
-from unittest import TestCase
+import unittest
-
 from dzclient import DatazillaResult
 
 
-
+class DatazillaResultTest(unittest.TestCase):
-class DatazillaResultTest(TestCase):
     def test_init_with_results(self):
         """Can initialize with a dictionary of results."""
         data = {"suite": {"test": [1, 2, 3]}}
@@ -54,3 +52,6 @@ def test_join_results(self):
                 "suite2": {"test2a": [4]},
                 }
             )
+
+if __name__ == '__main__':
+    unittest.main()