# HG changeset patch
# User Frederic Bastien <bastienf@iro.umontreal.ca>
# Date 1226699163 18000
# Node ID 317a052f9b140147ca4404f72666eae1bee0851a
# Parent  111e547ffa7b21140ede87c03849bb28a9a5ab5a
better main, allow to debug in a debugger.

diff -r 111e547ffa7b -r 317a052f9b14 _test_linear_regression.py
--- a/_test_linear_regression.py	Fri Nov 14 16:39:59 2008 -0500
+++ b/_test_linear_regression.py	Fri Nov 14 16:46:03 2008 -0500
@@ -21,5 +21,18 @@
         print 'mse = ',mse
         
 if __name__ == '__main__':
-    unittest.main()
-        
+    import sys
+
+    if len(sys.argv)==1:
+        unittest.main()
+    else:
+        assert sys.argv[1]=="--debug"
+        tests = []
+        for arg in sys.argv[2:]:
+            tests.append(arg)
+        if tests:
+            unittest.TestSuite(map(T_DataSet, tests)).debug()
+        else:
+            module = __import__("_test_linear_regression")
+            tests = unittest.TestLoader().loadTestsFromModule(module)
+            tests.debug()