diff --git a/test/integration/test_i_aws_encrytion_sdk_client.py b/test/integration/test_i_aws_encrytion_sdk_client.py index 639e21e95..56b0536fd 100644 --- a/test/integration/test_i_aws_encrytion_sdk_client.py +++ b/test/integration/test_i_aws_encrytion_sdk_client.py @@ -13,7 +13,6 @@ """Integration test suite for `aws_encryption_sdk`.""" import io import logging -import unittest import pytest from botocore.exceptions import BotoCoreError @@ -69,8 +68,9 @@ def test_remove_bad_client(): assert not test._regional_clients -class TestKMSThickClientIntegration(unittest.TestCase): - def setUp(self): +class TestKMSThickClientIntegration(object): + @pytest.fixture(autouse=True) + def apply_fixtures(self): self.kms_master_key_provider = setup_kms_master_key_provider() def test_encryption_cycle_default_algorithm_framed_stream(self):