lint

Author: lucasmcdonald3

test/unit/test_streaming_client_configs.py

@@ -33,10 +33,7 @@
 # Ideally, this logic would be based on mocking imports and testing logic,
 # but doing that introduces errors that cause other tests to fail.
 try:
-    from aws_cryptographic_materialproviders.mpl.references import (
-        ICryptographicMaterialsManager,
-        IKeyring,
-    )
+    from aws_cryptographic_materialproviders.mpl.references import ICryptographicMaterialsManager, IKeyring
     HAS_MPL = True
 
     from aws_encryption_sdk.materials_managers.mpl.cmm import CryptoMaterialsManagerFromMPL

test/unit/test_streaming_client_stream_decryptor.py

@@ -1071,7 +1071,7 @@ def test_GIVEN_materials_has_required_encryption_context_keys_attr_WHEN_read_hea
         self,
         mock_verifier,
         *_
-    ): 
+    ):
         required_encryption_context_keys_values = [
             # Case of empty encryption context list is not allowed;
             # if a list is provided, it must be non-empty.
@@ -1126,7 +1126,7 @@ def test_GIVEN_materials_has_required_encryption_context_keys_attr_WHEN_read_hea
                 test_decryptor._read_header()
 
                 # Then: Assert correctness of partitioned EC
-                for k in encryption_context:
+                for k in encryption_context.keys():
                     # If a key is in required_encryption_context_keys, then ...
                     if k in required_encryption_context_keys:
                         # ... its EC is in the StreamEncryptor._required_encryption_context

test/unit/test_streaming_client_stream_encryptor.py

@@ -512,7 +512,7 @@ def test_GIVEN_has_mpl_AND_encryption_materials_has_required_EC_keys_WHEN_prep_m
                 test_encryptor._prep_message()
 
                 # Then: Assert correctness of partitioned EC
-                for k in encryption_context:
+                for k in encryption_context.keys():
                     # If a key is in required_encryption_context_keys, then
                     if k in required_encryption_context_keys:
                         # 1) Its EC is in the StreamEncryptor._required_encryption_context