test_suite_md: improve md_to_from_psa() test function and related data

Signed-off-by: Valerio Setti <valerio.setti@nordicsemi.no>
diff --git a/tests/suites/test_suite_md.function b/tests/suites/test_suite_md.function
index 2a885e2..0a8e421 100644
--- a/tests/suites/test_suite_md.function
+++ b/tests/suites/test_suite_md.function
@@ -4,10 +4,6 @@
 
 #include "mbedtls/oid.h"
 #include "mbedtls/asn1.h"
-
-#define MD_PSA(md, psa) \
-    TEST_EQUAL(mbedtls_md_psa_alg_from_type(md), psa);  \
-    TEST_EQUAL(mbedtls_md_type_from_psa_alg(psa), md);
 /* END_HEADER */
 
 /* BEGIN_DEPENDENCIES
@@ -63,23 +59,13 @@
 /* END_CASE */
 
 /* BEGIN_CASE depends_on:MBEDTLS_PSA_CRYPTO_C */
-void md_to_from_psa()
+void md_to_from_psa(int md_alg_arg, int psa_alg_arg)
 {
-    /* We use a simplified implementation that relies on numerical values
-     * being aligned, so make sure they remain so. */
-    MD_PSA(MBEDTLS_MD_MD5, PSA_ALG_MD5);
-    MD_PSA(MBEDTLS_MD_RIPEMD160, PSA_ALG_RIPEMD160);
-    MD_PSA(MBEDTLS_MD_SHA1, PSA_ALG_SHA_1);
-    MD_PSA(MBEDTLS_MD_SHA224, PSA_ALG_SHA_224);
-    MD_PSA(MBEDTLS_MD_SHA256, PSA_ALG_SHA_256);
-    MD_PSA(MBEDTLS_MD_SHA384, PSA_ALG_SHA_384);
-    MD_PSA(MBEDTLS_MD_SHA512, PSA_ALG_SHA_512);
-    MD_PSA(MBEDTLS_MD_SHA3_224, PSA_ALG_SHA3_224);
-    MD_PSA(MBEDTLS_MD_SHA3_256, PSA_ALG_SHA3_256);
-    MD_PSA(MBEDTLS_MD_SHA3_384, PSA_ALG_SHA3_384);
-    MD_PSA(MBEDTLS_MD_SHA3_512, PSA_ALG_SHA3_512);
+    mbedtls_md_type_t md_alg = md_alg_arg;
+    psa_algorithm_t psa_alg = psa_alg_arg;
 
-    /* Don't test for NONE<->NONE as this is not guaranteed */
+    TEST_EQUAL(mbedtls_md_psa_alg_from_type(md_alg), psa_alg);  \
+    TEST_EQUAL(mbedtls_md_type_from_psa_alg(psa_alg), md_alg);
 }
 /* END_CASE */