Fix code style pt2

Signed-off-by: Elena Uziunaite <elena.uziunaite@arm.com>

scripts/generate_config_tests.py

diff --git a/scripts/generate_config_tests.py b/scripts/generate_config_tests.py
index 7d6b3c9..469d895 100755
--- a/scripts/generate_config_tests.py
+++ b/scripts/generate_config_tests.py
@@ -88,7 +88,7 @@
             return 'MBEDTLS_CIPHER_C:MBEDTLS_CIPHER_MODE_CBC'
         if name.startswith('MBEDTLS_PK_PARSE_EC_'):
             return 'MBEDTLS_PK_C:' + test_case.psa_or_3_6_feature_macro(
-                'PSA_KEY_TYPE_ECC_PUBLIC_KEY', test_case.Domain_3_6.USE_PSA)
+                'PSA_KEY_TYPE_ECC_PUBLIC_KEY', test_case.Domain36.USE_PSA)
 
         # For TLS settings, insist on having them once off and once on in
         # a configuration where both client support and server support are

scripts/generate_pkcs7_tests.py

diff --git a/scripts/generate_pkcs7_tests.py b/scripts/generate_pkcs7_tests.py
index 629b3a8..b97cebb 100755
--- a/scripts/generate_pkcs7_tests.py
+++ b/scripts/generate_pkcs7_tests.py
@@ -38,7 +38,8 @@
     Take in test_suite_pkcs7.data file.
     Allow for new tests to be added.
     """
-    mandatory_dep = test_case.psa_or_3_6_feature_macro("PSA_ALG_SHA_256", test_case.Domain_3_6.USE_PSA)
+    mandatory_dep = test_case.psa_or_3_6_feature_macro("PSA_ALG_SHA_256",
+                                                       test_case.Domain36.USE_PSA)
 
     test_name = "PKCS7 Parse Failure Invalid ASN1"
     test_function = "pkcs7_asn1_fail:"

scripts/mbedtls_framework/test_case.py

diff --git a/scripts/mbedtls_framework/test_case.py b/scripts/mbedtls_framework/test_case.py
index 276eba5..d2de727 100644
--- a/scripts/mbedtls_framework/test_case.py
+++ b/scripts/mbedtls_framework/test_case.py
@@ -33,7 +33,7 @@
     "PSA_KEY_TYPE_ECC_PUBLIC_KEY" : "MBEDTLS_PK_HAVE_ECC_KEYS"
 }
 
-class Domain_3_6(Enum):
+class Domain36(Enum):
     PSA = 1
     TLS_1_3_ONLY = 2
     USE_PSA = 3
@@ -117,18 +117,19 @@
     os.replace(tempfile, filename)
 
 def psa_or_3_6_feature_macro(psa_name: str,
-                             domain_3_6: Domain_3_6) -> str:
+                             domain_3_6: Domain36) -> str:
     """Determine the dependency symbol for a given psa_alg based on
        the domain and Mbed TLS version. For more information about the domains,
        and MBEDTLS_MD_CAN_ prefixed symbols, see transition-guards.md.
        Currently works with hashes and PK symbols only.
     """
 
-    if domain_3_6 == Domain_3_6.PSA or domain_3_6 == Domain_3_6.TLS_1_3_ONLY or not build_tree.is_mbedtls_3_6():
+    if domain_3_6 == Domain36.PSA or domain_3_6 == Domain36.TLS_1_3_ONLY or \
+        not build_tree.is_mbedtls_3_6():
         if psa_name in PK_MACROS_3_6 or psa_name in HASHES_3_6:
             return psa_information.psa_want_symbol(psa_name)
 
-    if domain_3_6 == Domain_3_6.USE_PSA:
+    if domain_3_6 == Domain36.USE_PSA:
         if psa_name in PK_MACROS_3_6:
             return PK_MACROS_3_6[psa_name]