Enabled terminal output when using set_trace() inside test cases run …

…by nosetests

README.txt

@@ -15,3 +15,4 @@ Credits
 -------
 
    * patch from Jean Jordaan for correct setup of colors
+   * patch from Antti Kaihola for nose compatibility

ipdb/__init__.py

@@ -6,6 +6,15 @@
 shell = IPShell(argv=[''])
 
 def set_trace():
+    try:
+        import nose.core
+        from traceback import extract_stack
+        nose_core_path = nose.core.__file__.rstrip('c')
+        if nose_core_path in (entry[0] for entry in extract_stack()):
+            from IPython.Shell import Term
+            Term.cout = sys.stdout = sys.__stdout__
+    except ImportError:
+        pass
     ip = ipapi.get()
     def_colors = ip.options.colors
     Pdb(def_colors).set_trace(sys._getframe().f_back)

setup.py

@@ -19,6 +19,10 @@
           # -*- Extra requirements: -*-
           'ipython',
       ],
+      extras_require={
+          # -*- Requirements for nose compatibility: -*-
+          'nosetests': ['nose'],
+      },
       entry_points="""
       # -*- Entry points: -*-
       """,