diff --git a/plotly/tests/test_core/test_file/test_file.py b/plotly/tests/test_core/test_file/test_file.py index 1e5def132fd..4eaa14a2f67 100644 --- a/plotly/tests/test_core/test_file/test_file.py +++ b/plotly/tests/test_core/test_file/test_file.py @@ -5,47 +5,47 @@ A module intended for use with Nose. """ - -from nose.tools import raises -from nose import with_setup - import random import string -import requests +from unittest import TestCase import plotly.plotly as py -import plotly.tools as tls from plotly.exceptions import PlotlyRequestError -def _random_filename(): - random_chars = [random.choice(string.ascii_uppercase) for _ in range(5)] - unique_filename = 'Valid Folder'+''.join(random_chars) - return unique_filename - - -def init(): - py.sign_in('PythonTest', '9v9f20pext') - +class FolderAPITestCase(TestCase): -@with_setup(init) -def test_create_folder(): - py.file_ops.mkdirs(_random_filename()) + def setUp(self): + py.sign_in('PythonTest', '9v9f20pext') + def _random_filename(self): + random_chars = [random.choice(string.ascii_uppercase) + for _ in range(5)] + unique_filename = 'Valid Folder'+''.join(random_chars) + return unique_filename -@with_setup(init) -def test_create_nested_folders(): - first_folder = _random_filename() - nested_folder = '{0}/{1}'.format(first_folder, _random_filename()) - py.file_ops.mkdirs(nested_folder) + def test_create_folder(self): + try: + py.file_ops.mkdirs(self._random_filename()) + except PlotlyRequestError as e: + self.fail('Expected this *not* to fail! Status: {}' + .format(e.status_code)) + def test_create_nested_folders(self): + first_folder = self._random_filename() + nested_folder = '{0}/{1}'.format(first_folder, self._random_filename()) + try: + py.file_ops.mkdirs(nested_folder) + except PlotlyRequestError as e: + self.fail('Expected this *not* to fail! Status: {}' + .format(e.status_code)) -@with_setup(init) -def test_duplicate_folders(): - first_folder = _random_filename() - py.file_ops.mkdirs(first_folder) - try: + def test_duplicate_folders(self): + first_folder = self._random_filename() py.file_ops.mkdirs(first_folder) - except PlotlyRequestError as e: - if e.status_code != 409: - raise e + try: + py.file_ops.mkdirs(first_folder) + except PlotlyRequestError as e: + self.assertTrue(400 <= e.status_code < 500) + else: + self.fail('Expected this to fail!') diff --git a/plotly/tests/test_optional/test_utils/test_utils.py b/plotly/tests/test_optional/test_utils/test_utils.py index db3d8dc3f56..051230a9228 100644 --- a/plotly/tests/test_optional/test_utils/test_utils.py +++ b/plotly/tests/test_optional/test_utils/test_utils.py @@ -225,7 +225,7 @@ def test_pandas_json_encoding(): # Test that data wasn't mutated assert_series_equal(df['col 1'], pd.Series([1, 2, 3, dt(2014, 1, 5), - pd.NaT, np.NaN, np.Inf])) + pd.NaT, np.NaN, np.Inf], name='col 1')) j2 = json.dumps(df.index, cls=utils.PlotlyJSONEncoder) assert(j2 == '[0, 1, 2, 3, 4, 5, 6]')