commit eb8c665a5337443dd516e0b8ab67e93c8ea3919b
author Ryan Everett <ryan.everett@arm.com> Wed Feb 07 17:25:39 2024 +0000
committer Ryan Everett <ryan.everett@arm.com> Wed Feb 07 17:32:16 2024 +0000
tree 0db15990470475d3f656f143225eaa8c523106ba
parent 5d2e82f0cecf85b347a7d170e55597c58226118c

Reformat wrapper generation code

Signed-off-by: Ryan Everett <ryan.everett@arm.com>

tests/scripts/generate_psa_wrappers.py

diff --git a/tests/scripts/generate_psa_wrappers.py b/tests/scripts/generate_psa_wrappers.py
index 816dd5f..9ca6b29 100755
--- a/tests/scripts/generate_psa_wrappers.py
+++ b/tests/scripts/generate_psa_wrappers.py
@@ -145,7 +145,8 @@
         # Proof-of-concept: just instrument one function for now
         if function_name == 'psa_cipher_encrypt':
             return True
-        if function_name == 'psa_key_derivation_output_bytes' or function_name == 'psa_key_derivation_input_bytes':
+        if function_name in ('psa_key_derivation_output_bytes',
+                             'psa_key_derivation_input_bytes'):
             return True
         if function_name in ('psa_import_key',
                              'psa_export_key',