derive_output test: add other key type value 11 to handle raw key type

Signed-off-by: Przemek Stekiel <przemyslaw.stekiel@mobica.com>
diff --git a/tests/suites/test_suite_psa_crypto.function b/tests/suites/test_suite_psa_crypto.function
index accd4b3..45f2c67 100644
--- a/tests/suites/test_suite_psa_crypto.function
+++ b/tests/suites/test_suite_psa_crypto.function
@@ -6998,13 +6998,14 @@
                                                                     inputs[i]->len ),
                                     statuses[i] );
                         break;
-                    case 1: // input key
+                    case 1: // input key, type DERIVE
+                    case 11: // input key, type RAW
                         psa_set_key_usage_flags( &attributes2, PSA_KEY_USAGE_DERIVE );
                         psa_set_key_algorithm( &attributes2, alg );
                         psa_set_key_type( &attributes2, PSA_KEY_TYPE_DERIVE );
 
                         // other secret of type RAW_DATA passed with input_key
-                        if( statuses[i] == PSA_ERROR_INVALID_ARGUMENT )
+                        if( other_key_input_type == 11 )
                             psa_set_key_type( &attributes2, PSA_KEY_TYPE_RAW_DATA );
 
                         PSA_ASSERT( psa_import_key( &attributes2,