def test_print_stack_default(self): original_stderr = sys.__stderr__ stderr = sys.stderr = StringIO.StringIO() try: stacktrace.print_stack() finally: sys.stderr = original_stderr self.assertEndsWith(stderr.getvalue(), 'stacktrace.print_stack()\n')
def test_print_stack_options(self): f = StringIO.StringIO() frame = get_frame() stacktrace.print_stack(f=frame, limit=100, file=f) self.assertEndsWith(f.getvalue(), 'return sys._getframe()\n') self.assertTrue(len(f.getvalue().split('\n')) > 4) f = StringIO.StringIO() stacktrace.print_stack(f=frame, limit=2, file=f) self.assertEqual(4, len(f.getvalue().strip().split('\n'))) self.assertEndsWith(f.getvalue(), 'return sys._getframe()\n')