Fixed color initialization and koans filtering bugs for windows users

gregmalcolm · gregmalcolm · commit 0dd9acbe01fb · 2010-10-03T02:31:37.000-04:00
diff --git a/python 2/runner/sensei.py b/python 2/runner/sensei.py
@@ -10,6 +10,7 @@
 from runner import path_to_enlightenment
 
 from libs.colorama import init, Fore, Style
+init() # init colorama
 
 class Sensei(MockableTestResult):
     def __init__(self, stream):
@@ -239,6 +240,8 @@ def total_koans(self):
     def filter_all_lessons(self):
         if not self.all_lessons:
             self.all_lessons = glob.glob('koans/about*.py')
-            self.all_lessons.remove('koans/about_extra_credit.py')
-
+            self.all_lessons = filter(lambda filename:
+                                      "about_extra_credit" not in filename,
+                                      self.all_lessons)
+            
         return self.all_lessons
diff --git a/python 3/runner/sensei.py b/python 3/runner/sensei.py
@@ -10,6 +10,7 @@
 from runner import path_to_enlightenment
 
 from libs.colorama import init, Fore, Style
+init() # init colorama
 
 class Sensei(MockableTestResult):
     def __init__(self, stream):
@@ -239,6 +240,8 @@ def total_koans(self):
     def filter_all_lessons(self):
         if not self.all_lessons:
             self.all_lessons = glob.glob('koans/about*.py')
-            self.all_lessons.remove('koans/about_extra_credit.py')
+            self.all_lessons = list(filter(lambda filename:
+                                      "about_extra_credit" not in filename,
+                                      self.all_lessons))
 
-        return self.all_lessons      
+        return self.all_lessons     
