Fix the color output for Windows.

coderanger · coderanger · commit e01e8fb3d384 · 2010-08-11T15:47:58.000-07:00
diff --git a/pyzen/runner.py b/pyzen/runner.py
@@ -30,7 +30,9 @@ def addFailure(self, test, err):
     def printErrorList(self, flavour, errors):
             for test, err in errors:
                 self.stream.writeln(self.separator1)
-                self.stream.writeln("%s%s: %s%s" % (COLOR_FAIL,flavour,self.getDescription(test),COLOR_RESET))
+                self.stream.write(COLOR_FAIL)
+                self.stream.writeln("%s: %s" % (flavour,self.getDescription(test)))
+                self.stream.write(COLOR_RESET)
                 self.stream.writeln(self.separator2)
                 self.stream.writeln("%s" % err)
 
@@ -67,9 +69,11 @@ def run(self, test):
             if errored:
                 if failed: self.stream.write(", ")
                 self.stream.write("errors=%d" % errored)
-            self.stream.writeln(")"+COLOR_RESET)
+            self.stream.writeln(")")
         else:
-            self.stream.writeln(COLOR_SUCCESS+"OK"+COLOR_RESET)
+            self.stream.write(COLOR_SUCCESS)
+            self.stream.writeln("OK")
+        self.stream.write(COLOR_RESET)
         return result
 
 def get_test_runner(nocolor):
