Allow running certain tests separately

E.g. `python2 -m unittest tests.MakePatchTestCase.test_objects`
or `nose tests:MakePatchTestCase.test_objects`.

1 files changed, 19 insertions, 19 deletions

diff --git a/tests.py b/tests.py
index 5cbff33..24f3908 100755
--- a/tests.py
+++ b/tests.py
@@ -414,30 +414,30 @@ class ConflictTests(unittest.TestCase):
         self.assertRaises(jsonpatch.JsonPatchConflict, jsonpatch.apply_patch, src, patch_obj)
 
 
+if __name__ == '__main__':
+    modules = ['jsonpatch']
 
-modules = ['jsonpatch']
 
+    def get_suite():
+        suite = unittest.TestSuite()
+        suite.addTest(doctest.DocTestSuite(jsonpatch))
+        suite.addTest(unittest.makeSuite(ApplyPatchTestCase))
+        suite.addTest(unittest.makeSuite(EqualityTestCase))
+        suite.addTest(unittest.makeSuite(MakePatchTestCase))
+        suite.addTest(unittest.makeSuite(InvalidInputTests))
+        suite.addTest(unittest.makeSuite(ConflictTests))
+        return suite
 
-def get_suite():
-    suite = unittest.TestSuite()
-    suite.addTest(doctest.DocTestSuite(jsonpatch))
-    suite.addTest(unittest.makeSuite(ApplyPatchTestCase))
-    suite.addTest(unittest.makeSuite(EqualityTestCase))
-    suite.addTest(unittest.makeSuite(MakePatchTestCase))
-    suite.addTest(unittest.makeSuite(InvalidInputTests))
-    suite.addTest(unittest.makeSuite(ConflictTests))
-    return suite
 
+    suite = get_suite()
 
-suite = get_suite()
+    for module in modules:
+        m = __import__(module, fromlist=[module])
+        suite.addTest(doctest.DocTestSuite(m))
 
-for module in modules:
-    m = __import__(module, fromlist=[module])
-    suite.addTest(doctest.DocTestSuite(m))
+    runner = unittest.TextTestRunner(verbosity=1)
 
-runner = unittest.TextTestRunner(verbosity=1)
+    result = runner.run(suite)
 
-result = runner.run(suite)
-
-if not result.wasSuccessful():
-    sys.exit(1)
+    if not result.wasSuccessful():
+        sys.exit(1)