Refactoring tests

Author: leandrodamascena

tests/functional/data_masking/required_dependencies/test_erase_data_masking.py

@@ -321,6 +321,7 @@ def test_mask_nested_field_success(data_masker):
     # THEN the nested field should be masked while other data remains unchanged
     assert data_masked == {"user": {"contact": {"details": {"address": {"street": "123 Main St", "zip": "xxx"}}}}}
 
+
 def test_erase_dictionary_with_masking_rules(data_masker):
     # GIVEN a dictionary with nested sensitive data
     data = {"user": {"name": "John Doe", "ssn": "123-45-6789", "address": {"street": "123 Main St", "zip": "12345"}}}
@@ -411,6 +412,7 @@ def test_erase_handles_empty_string_with_dynamic_mask(data_masker):
     # THEN empty string should be returned
     assert result == ""
 
+
 def test_erase_dictionary_with_masking_rules_wrong_field(data_masker):
     # GIVEN a dictionary with nested sensitive data
     data = {"user": {"name": "John Doe", "ssn": "123-45-6789", "address": {"street": "123 Main St", "zip": "12345"}}}

tests/unit/data_masking/required_dependencies/test_base_functions.py

@@ -1,20 +1,13 @@
-
-import json
-
 import pytest
 
 from aws_lambda_powertools.utilities.data_masking.base import DataMasking
-from aws_lambda_powertools.utilities.data_masking.constants import DATA_MASKING_STRING
-from aws_lambda_powertools.utilities.data_masking.exceptions import (
-    DataMaskingFieldNotFoundError,
-    DataMaskingUnsupportedTypeError,
-)
 
 
 @pytest.fixture
 def data_masker() -> DataMasking:
     return DataMasking()
 
+
 def test_mask_nested_field_with_non_dict_value(data_masker):
     # GIVEN nested data where a middle path component is not a dictionary
     data = {"user": {"contact": "not_a_dict", "details": {"ssn": "123-45-6789"}}}  # This will stop the traversal