rewrote test

Author: Kully

plotly/tests/test_core/test_offline/test_offline.py

@@ -6,6 +6,7 @@
 
 from nose.tools import raises
 from unittest import TestCase
+from plotly.tests.utils import PlotlyTestCase
 import json
 
 import plotly
@@ -21,6 +22,7 @@
 
 PLOTLYJS = plotly.offline.offline.get_plotlyjs()
 
+
 class PlotlyOfflineTestCase(TestCase):
     def setUp(self):
         pass
@@ -79,11 +81,22 @@ def test_autoresizing(self):
         for resize_code_string in resize_code_strings:
             self.assertTrue(resize_code_string not in html)
 
-    def test_plot_rendered_if_non_plotly_domain(self):
+
+class PlotlyOfflineOtherDomainTestCase(PlotlyTestCase):
+    def setUp(self):
+        super(PlotlyOfflineOtherDomainTestCase, self).setUp()
         plotly.tools.set_config_file(plotly_domain='https://stage.plot.ly',
                                      plotly_api_domain='https://api-stage.plot.ly')
+        plotly.plotly.sign_in('PlotlyTestShark', 'kI2H0E7sTZ5RM9yoaK4X')
 
+    def test_plot_rendered_if_non_plotly_domain(self):
         html = plotly.offline.plot(fig, output_type='div')
+
         # test that 'Export to stage.plot.ly' is in the html
         self.assertTrue('Export to stage.plot.ly' in html)
-        self.assertTrue('Export to plot.ly' not in html)
+
+    def tearDown(self):
+        plotly.tools.set_config_file(plotly_domain='https://plot.ly',
+                                     plotly_api_domain='https://api.plot.ly')
+        plotly.plotly.sign_in('PythonTest', '9v9f20pext')
+        super(PlotlyOfflineOtherDomainTestCase, self).tearDown()