scripts: generate_test_keys: minor improvements

- better guard for the "ecp.h" header file;
- use already existing Mbed TLS' macro for unused variables and place it
  at the beginning of the declaration of "predefined_keys".

Signed-off-by: Valerio Setti <vsetti@baylibre.com>
diff --git a/scripts/generate_test_keys.py b/scripts/generate_test_keys.py
index 78551c9..7cec7b3 100755
--- a/scripts/generate_test_keys.py
+++ b/scripts/generate_test_keys.py
@@ -100,9 +100,11 @@
 #ifndef TEST_TEST_KEYS_H
 #define TEST_TEST_KEYS_H
 
-#if !defined(MBEDTLS_VERSION_MAJOR) || MBEDTLS_VERSION_MAJOR >= 4
+#if TF_PSA_CRYPTO_VERSION_MAJOR >= 1
+#include <tf_psa_crypto_common.h>
 #include <mbedtls/private/ecp.h>
 #else
+#include <common.h>
 #include <mbedtls/ecp.h>
 #endif
 
@@ -118,13 +120,7 @@
     size_t pub_key_len;
 }};
 
-#if defined(__GNUC__) || defined(__clang__)
-#define UNUSED __attribute__((unused))
-#else
-#define UNUSED
-#endif
-
-static struct predefined_key_element predefined_keys[] UNUSED = {{
+MBEDTLS_MAYBE_UNUSED static struct predefined_key_element predefined_keys[] = {{
 {}
 }};