Provide what is *not* in 'six' already You should just do "import ...backport" and then use the new imports as six.moves.<name> """ __all__ = ['utf8_cat'] from six import PY3 from six import MovedModule from six import MovedAttribute from six import add_move attributes = [ MovedModule('xmlrpc_client', 'xmlrpclib', 'xmlrpc.client'), MovedAttribute('HTTPError', 'urllib2', 'urllib.error'), MovedAttribute('URLError', 'urllib2', 'urllib.error'), MovedAttribute('Request', 'urllib2', 'urllib.request'), MovedAttribute('urlopen', 'urllib2', 'urllib.request'), MovedAttribute('urllib_version', 'urllib2', 'urllib.request', '__version__'), MovedAttribute('urlparse', 'urlparse', 'urllib.parse'), ] for attr in attributes: add_move(attr) del attr del attributes if PY3: def utf8_cat(f): """Read the given file in utf8 encoding"""
# -------------------------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. # Licensed under the MIT License. See License.txt in the project root for license information. # -------------------------------------------------------------------------------------------- from azure.iot.hub.devicesdk.internal_client import InternalClient from azure.iot.hub.devicesdk.device_client import DeviceClient from azure.iot.hub.devicesdk.module_client import ModuleClient from azure.iot.hub.devicesdk.auth.authentication_provider_factory import from_connection_string from azure.iot.hub.devicesdk.transport.mqtt.mqtt_transport import MQTTTransport import pytest from six import add_move, MovedModule add_move(MovedModule("mock", "mock", "unittest.mock")) from six.moves import mock from mock import MagicMock connection_string_format = "HostName={};DeviceId={};SharedAccessKey={}" shared_access_key = "Zm9vYmFy" hostname = "beauxbatons.academy-net" device_id = "MyPensieve" @pytest.fixture def connection_string(): connection_string = connection_string_format.format( hostname, device_id, shared_access_key) return connection_string
from gslib.utils import posix_util from gslib.utils import system_util from gslib.utils import hashing_helper from gslib.utils.copy_helper import _DelegateUploadFileToObject from gslib.utils.copy_helper import _GetPartitionInfo from gslib.utils.copy_helper import _SelectUploadCompressionStrategy from gslib.utils.copy_helper import _SetContentTypeFromFile from gslib.utils.copy_helper import ExpandUrlToSingleBlr from gslib.utils.copy_helper import FilterExistingComponents from gslib.utils.copy_helper import GZIP_ALL_FILES from gslib.utils.copy_helper import PerformParallelUploadFileToObjectArgs from gslib.utils.copy_helper import WarnIfMvEarlyDeletionChargeApplies from six import add_move, MovedModule add_move(MovedModule('mock', 'mock', 'unittest.mock')) from six.moves import mock _CalculateB64EncodedMd5FromContents = ( hashing_helper.CalculateB64EncodedMd5FromContents) class TestCpFuncs(GsUtilUnitTestCase): """Unit tests for parallel upload functions in cp command.""" def testGetPartitionInfo(self): """Tests the _GetPartitionInfo function.""" # Simplest case - threshold divides file_size. (num_components, component_size) = _GetPartitionInfo(300, 200, 10) self.assertEqual(30, num_components) self.assertEqual(10, component_size)
import sys from puck.backend import (PYPI_URL, echo_backend, high_version_backend, pypi_backend) from six import MovedModule, add_move import pytest add_move(MovedModule('mock', 'mock', 'unittest.mock')) # isort:skip from six.moves import mock # noqa isort:skip def test_high_version_backend(): assert high_version_backend('foo') == '99.99.99' def test_echo_backend(): assert echo_backend('foo') == 'foo' @pytest.mark.skipif(sys.version_info < (3, 3), reason="requires python3.3") @mock.patch('xmlrpc.client.ServerProxy') def test_pypi_backend_py3(server_mock): expected = None pypi_mock = server_mock.return_value pypi_mock.package_releases.return_value = [expected] return_val = pypi_backend('foo') server_mock.assert_called_with(PYPI_URL) pypi_mock.package_releases.assert_called_once()