TestCase for the :habit: hamster, this one's for you @theengineear

chriddyp · chriddyp · commit 3d95463d5b7b · 2015-07-02T14:58:23.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
@@ -5,14 +5,16 @@
 from __future__ import absolute_import
 
 import os
+from unittest import TestCase
 
 import plotly
 
 
-def test_downloading_file_saves_it_to_the_disk():
-    dummy_js_url = ('https://gist.githubusercontent.com/chriddyp/'
-                    'f40bd33d1eab6f0715dc/raw/'
-                    '24cd2e4e62ceea79e6e790b3a2c94cda63510ede/test.js')
+class PlotlyOfflineTestCase(TestCase):
+    def test_downloading_file_saves_it_to_the_disk(self):
+        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
+        plotly.offline.download_plotlyjs(dummy_js_url)
+        assert os.path.isfile(plotly.offline.PLOTLY_OFFLINE_BUNDLE) is True
diff --git a/plotly/tests/test_optional/test_offline/test_offline.py b/plotly/tests/test_optional/test_offline/test_offline.py
@@ -4,28 +4,27 @@
 """
 from __future__ import absolute_import
 from nose.tools import raises
+from unittest import TestCase
 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]}])
+class PlotlyOfflineTestCase(TestCase):
+    @raises(PlotlyError)
+    def test_calling_iplot_before_initializing_raises_an_error(self):
+        plotly.offline.iplot([{'x': [1, 2, 3]}])
 
+    def test_no_errors_are_raised_when_initializing_offline_mode(self):
+        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
+    @raises(PlotlyError)
+    def test_initializing_before_downloading_raises_an_error(self):
+        try:
+            os.remove(plotly.offline.PLOTLY_OFFLINE_BUNDLE)
+        except OSError:
+            pass
 
-    plotly.offline.init_notebook_mode()
+        plotly.offline.init_notebook_mode()
