fixup (pyenv environ var)

etpinard · etpinard · commit 3239fcbb68aa · 2016-08-25T13:28:39.000-04:00
diff --git a/plotly/tests/test_core/test_jupyter/test_jupyter.py b/plotly/tests/test_core/test_jupyter/test_jupyter.py
@@ -14,16 +14,16 @@
 PATH_FIXTURES = path.join(PATH_ROOT, 'fixtures')
 PATH_TEST_NB = path.join(PATH_FIXTURES, 'test.ipynb')
 PATH_TEST_HTML = path.join(PATH_FIXTURES, 'test.html')
-PYENV_VERSION = environ['PYENV_VERSION']
 
 
 class PlotlyJupyterTestCase(TestCase):
     def setUp(self):
         with open(PATH_TEST_NB, 'r') as f:
             self.nb = nbformat.read(f, as_version=4)
 
-        if PYENV_VERSION:
-            self.ep = ExecutePreprocessor(timeout=600, kernel_name=PYENV_VERSION)
+        if 'PYENV_VERSION' in environ:
+            kernel_name = environ['PYENV_VERSION']
+            self.ep = ExecutePreprocessor(timeout=600, kernel_name=kernel_name)
         else:
             self.ep = ExecutePreprocessor(timeout=600)
 
