Merge pull request gregmalcolm#48 from orzrd/master

gregmalcolm · gregmalcolm · commit 11414c511779 · 2013-04-28T15:49:19.000-07:00
fix sensei display bug
diff --git a/python2/runner/sensei.py b/python2/runner/sensei.py
@@ -4,6 +4,7 @@
 import unittest
 import re
 import sys
+import os
 import glob
 
 import helper
@@ -235,6 +236,7 @@ def say_something_zenlike(self):
 
     def total_lessons(self):
         all_lessons = self.filter_all_lessons()
+
         if all_lessons:
           return len(all_lessons)
         else:
@@ -244,10 +246,10 @@ def total_koans(self):
         return self.tests.countTestCases()
 
     def filter_all_lessons(self):
+        cur_dir = os.path.split(os.path.realpath(__file__))[0]
         if not self.all_lessons:
-            self.all_lessons = glob.glob('koans/about*.py')
+            self.all_lessons = glob.glob('{0}/../koans/about*.py'.format(cur_dir))
             self.all_lessons = filter(lambda filename:
                                       "about_extra_credit" not in filename,
                                       self.all_lessons)
-
         return self.all_lessons
diff --git a/python3/runner/sensei.py b/python3/runner/sensei.py
@@ -3,6 +3,7 @@
 
 import unittest
 import re
+import os
 import glob
 
 from . import helper
@@ -243,8 +244,9 @@ def total_koans(self):
         return self.tests.countTestCases()
 
     def filter_all_lessons(self):
+        cur_dir = os.path.split(os.path.realpath(__file__))[0]
         if not self.all_lessons:
-            self.all_lessons = glob.glob('koans/about*.py')
+            self.all_lessons = glob.glob('{0}/../koans/about*.py'.format(cur_dir))
             self.all_lessons = list(filter(lambda filename:
                                       "about_extra_credit" not in filename,
                                       self.all_lessons))
