diff --git a/test/unit/test_serialize.py b/test/unit/test_serialize.py index 98068c2ca..511048d80 100644 --- a/test/unit/test_serialize.py +++ b/test/unit/test_serialize.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 aws_encryption_sdk.internal.formatting.serialize""" -import unittest - import pytest from mock import MagicMock, patch, sentinel @@ -51,8 +49,9 @@ def test_serialize_frame_invalid_sequence_number(sequence_number, error_message) excinfo.match(error_message) -class TestSerialize(unittest.TestCase): - def setUp(self): +class TestSerialize(object): + @pytest.fixture(autouse=True) + def apply_fixtures(self): self.mock_algorithm = MagicMock() self.mock_algorithm.encryption_algorithm.block_size = VALUES["block_size"] self.mock_algorithm.algorithm_id = VALUES["algorithm_id"] @@ -84,8 +83,8 @@ def setUp(self): self.mock_signer = MagicMock() self.mock_signer.update.return_value = None self.mock_signer.finalize.return_value = VALUES["signature"] - - def tearDown(self): + yield + # Run tearDown self.mock_serialize_acc_patcher.stop() self.mock_encrypt_patcher.stop() self.mock_valid_frame_length_patcher.stop()