X-Git-Url: http://mj.ucw.cz/gitweb/?a=blobdiff_plain;f=t%2Fmoe%2Flog.py;h=592fb846b047bd7b1952c3e6866080aa1074b67d;hb=d8de72934717e8f287301c07d2ec70f1ca332e79;hp=56dbe8f1f8dfa92d87b354ab9c21e60e5445a834;hpb=5701572b35e7ebc7c2d96cc603cd720fb2649ac4;p=eval.git diff --git a/t/moe/log.py b/t/moe/log.py index 56dbe8f..592fb84 100644 --- a/t/moe/log.py +++ b/t/moe/log.py @@ -2,8 +2,11 @@ import sys import os +import traceback -progress_file = os.fdopen(os.dup(1), "w", 0) +orig_stdout_fd = os.dup(1) +orig_stderr_fd = os.dup(2) +progress_file = os.fdopen(orig_stdout_fd, "w", 0) class MoeLog: @@ -37,3 +40,8 @@ class MoeLog: self.log_file.flush() default = MoeLog() + +def fatal_exception(): + os.dup2(orig_stderr_fd, 2) + traceback.print_exc() + sys.exit(1)