refactor tests to handle core vs optional deps (ipython)

chriddyp · chriddyp · commit 27936092679f · 2015-06-30T17:42:41.000-04:00
diff --git a/plotly/tests/test_core/test_offline/test_offline.py b/plotly/tests/test_core/test_offline/test_offline.py
@@ -3,38 +3,16 @@
 
 """
 from __future__ import absolute_import
-from nose.tools import raises
-import os
 
+import os
 
-from plotly.exceptions import PlotlyError
 import plotly
 
 
-@raises(PlotlyError)
-def test_calling_iplot_before_initializing_raises_an_error():
-    plotly.offline.iplot([{'x': [1, 2, 3]}])
-
-
-def test_no_errors_are_raised_when_properly_initializing_offline_mode():
-    plotly.offline.init_notebook_mode()
-    plotly.offline.iplot([{'x': [1, 2, 3]}])
-
-
 def test_downloading_file_saves_it_to_the_disk():
     dummy_js_url = ('https://gist.githubusercontent.com/chriddyp/'
                     'f40bd33d1eab6f0715dc/raw/'
                     '24cd2e4e62ceea79e6e790b3a2c94cda63510ede/test.js')
 
     plotly.offline.download_plotlyjs(dummy_js_url)
     assert os.path.isfile(plotly.offline.PLOTLY_OFFLINE_BUNDLE) is True
-
-
-@raises(PlotlyError)
-def test_initializing_before_downloading_raises_an_error():
-    try:
-        os.remove(plotly.offline.PLOTLY_OFFLINE_BUNDLE)
-    except OSError:
-        pass
-
-    plotly.offline.init_notebook_mode()
diff --git a/plotly/tests/test_optional/test_offline/test_offline.py b/plotly/tests/test_optional/test_offline/test_offline.py
@@ -0,0 +1,31 @@
+"""
+test__offline
+
+"""
+from __future__ import absolute_import
+from nose.tools import raises
+import os
+
+
+from plotly.exceptions import PlotlyError
+import plotly
+
+
+@raises(PlotlyError)
+def test_calling_iplot_before_initializing_raises_an_error():
+    plotly.offline.iplot([{'x': [1, 2, 3]}])
+
+
+def test_no_errors_are_raised_when_properly_initializing_offline_mode():
+    plotly.offline.init_notebook_mode()
+    plotly.offline.iplot([{'x': [1, 2, 3]}])
+
+
+@raises(PlotlyError)
+def test_initializing_before_downloading_raises_an_error():
+    try:
+        os.remove(plotly.offline.PLOTLY_OFFLINE_BUNDLE)
+    except OSError:
+        pass
+
+    plotly.offline.init_notebook_mode()
