%cpaste: strip whitespace before >>> in regex to allow pasting doctests

test/test_cpaste.ipy

@@ -1,61 +1,62 b''
 """Test cpaste magic"""
 
 tests = {'pass': ["> > > run()",
                   ">>> > run()",
                   "+++ run()",
-                  "++ run()"],
+                  "++ run()",
+		  "  >>> run()"],
 
          'fail': ["+ + run()",
                   " ++ run()"]}
 
 from StringIO import StringIO
 import sys
 
 stdin_save = sys.stdin
 
 # NOTE: no blank lines allowed in function definition
 def testcase(code,should_fail=False):
     """Execute code via 'cpaste' and ensure it was executed, unless
     should_fail is set.
 
     """
     _ip.user_ns['code_ran'] = False
     #
     src = StringIO()
     src.write('\n')
     src.write(code)
     src.write('\n--\n')
     src.seek(0)
     #
     sys.stdin = src
     
     try:
         cpaste
     except:
         if not should_fail:
             raise AssertionError("Failure not expected : '%s'" %
                                  code)
     else:
         assert code_ran
         if should_fail:
             raise AssertionError("Failure expected : '%s'" % code)
     #
     finally:
         sys.stdin = stdin_save
     #
 
 def run():
     """Marker function: sets a flag when executed.
 
     """
     _ip.user_ns['code_ran'] = True
     return 'run' # return string so '+ run()' doesn't result in success
 
 
 ### Actual testing happens here
 
 for code in tests['pass']:
     testcase(code)
 
 for code in tests['fail']:
     testcase(code,should_fail=True)