unit/test_providers_raw_master_key.py

Author: Andres Sanchez

test/unit/test_providers_raw_master_key.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.
 """Test suite for aws_encryption_sdk.key_providers.raw.RawMasterKey"""
-import unittest
-
 import pytest
 from mock import MagicMock, patch, sentinel
 
@@ -27,8 +25,9 @@
 pytestmark = [pytest.mark.unit, pytest.mark.local]
 
 
-class TestRawMasterKey(unittest.TestCase):
-    def setUp(self):
+class TestRawMasterKey(object):
+    @pytest.fixture(autouse=True)
+    def apply_fixtures(self):
         self.mock_algorithm = MagicMock()
         self.mock_algorithm.__class__ = Algorithm
         self.mock_algorithm.data_key_len = sentinel.data_key_len