tch16

akki2825 · Sep 5, 2017 · 9e042d8 · 9e042d8
1 parent d41f7a3
commit 9e042d8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/code/tests/test_notebooks.py b/code/tests/test_notebooks.py
@@ -92,7 +92,6 @@ def test_ch09(self):
 
         # run only on Py3, because of the Py3 specific pickle files
         if (sys.version_info >= (3, 0)):
-            os.chdir(this_dir)
             run_ipynb(os.path.join(this_dir, '../ch09/ch09.ipynb'))
         else:
             pass
@@ -129,8 +128,10 @@ def test_ch15(self):
 
     def test_ch16(self):
         this_dir = os.path.dirname(os.path.abspath(__file__))
-        run_ipynb(os.path.join(this_dir,
-                               '../ch16/ch16.ipynb'))
+        # run only on Py3, because of Py2 unicode handling
+        if (sys.version_info >= (3, 0)):
+            run_ipynb(os.path.join(this_dir,
+                                   '../ch16/ch16.ipynb'))
 
 
 if __name__ == '__main__':