Migrate unit/test_aws_encryption_sdk.py from unittest to pytest

Author: Andres Sanchez

test/unit/test_aws_encryption_sdk.py

@@ -11,8 +11,6 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 """Unit test suite for high-level functions in aws_encryption_sdk module"""
-import unittest
-
 import pytest
 import six
 from mock import MagicMock, patch, sentinel
@@ -23,8 +21,9 @@
 pytestmark = [pytest.mark.unit, pytest.mark.local]
 
 
-class TestAwsEncryptionSdk(unittest.TestCase):
-    def setUp(self):
+class TestAwsEncryptionSdk(object):
+    @pytest.fixture(autouse=True)
+    def apply_fixtures(self):
         # Set up StreamEncryptor patch
         self.mock_stream_encryptor_patcher = patch("aws_encryption_sdk.StreamEncryptor")
         self.mock_stream_encryptor = self.mock_stream_encryptor_patcher.start()
@@ -77,5 +76,6 @@ def test_stream_decryptor_decrypt(self):
         assert test is self.mock_stream_decryptor_instance
 
     def test_stream_unknown(self):
-        with six.assertRaisesRegex(self, ValueError, "Unsupported mode: *"):
+        with pytest.raises(ValueError) as excinfo:
             aws_encryption_sdk.stream(mode="ERROR", a=sentinel.a, b=sentinel.b, c=sentinel.b)
+        excinfo.match("Unsupported mode: *")