[filter/nnfw] Unittest for invoke

Added unittest for invoke for nnfw tensor filter extension
Added unittest for get input/output dimensions
Corresponding bug fixes are also added

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>

diff --git a/ext/nnstreamer/tensor_filter/tensor_filter_nnfw.c b/ext/nnstreamer/tensor_filter/tensor_filter_nnfw.c
index 3073d1a..896493f 100644 (file)
--- a/ext/nnstreamer/tensor_filter/tensor_filter_nnfw.c
+++ b/ext/nnstreamer/tensor_filter/tensor_filter_nnfw.c
@@ -344,18 +344,21 @@ static int nnfw_tensor_memory_set (const GstTensorFilterProperties * prop,
   NNFW_STATUS nnfw_status;
   int err = 0;
   guint idx;
-  unsigned int num_tensors;
+  unsigned int num_tensors = 0;
 
   g_return_val_if_fail (prop != NULL, -EINVAL);
   g_return_val_if_fail (mem != NULL, -EINVAL);
+  g_return_val_if_fail (pdata != NULL, -EINVAL);
   g_return_val_if_fail (pdata->session != NULL, -EPERM);
 
-  if (is_input) {
-    g_return_val_if_fail (prop->input_configured == TRUE, -EPERM);
+  if (is_input == TRUE && prop->input_configured == TRUE)
     num_tensors = prop->input_meta.num_tensors;
-  } else {
-    g_return_val_if_fail (prop->output_configured == TRUE, -EPERM);
+  else if (is_input == FALSE && prop->output_configured == TRUE)
     num_tensors = prop->output_meta.num_tensors;
+  else {
+    GstTensorsInfo info;
+    err = nnfw_tensors_info_get (pdata, is_input, &info);
+    num_tensors = info.num_tensors;
   }
 
   for (idx = 0; idx < num_tensors; idx ++) {

diff --git a/tests/tizen_nnfw_runtime/unittest_tizen_nnfw_runtime_raw.cpp b/tests/tizen_nnfw_runtime/unittest_tizen_nnfw_runtime_raw.cpp
index ff87fda..d9e0599 100644 (file)
--- a/tests/tizen_nnfw_runtime/unittest_tizen_nnfw_runtime_raw.cpp
+++ b/tests/tizen_nnfw_runtime/unittest_tizen_nnfw_runtime_raw.cpp
@@ -87,6 +87,153 @@ TEST (nnstreamer_nnfw_runtime_raw_functions, open_close_01_n)
 }
 
 /**
+ * @brief Get input/output dimensions with nnfw subplugin
+ */
+TEST (nnstreamer_nnfw_runtime_raw_functions, get_dimension)
+{
+  int ret;
+  void *data = NULL;
+  gchar *test_model;
+  gchar *model_path;
+  const gchar *root_path = g_getenv ("NNSTREAMER_BUILD_ROOT_PATH");
+  GstTensorsInfo info, res;
+
+  ASSERT_NE (root_path, nullptr);
+
+  /** nnfw needs a directory with model file and metadata in that directory */
+  model_path = g_build_filename (root_path, "tests", "test_models", "models",
+      NULL);
+  GstTensorFilterProperties prop = {
+    .fwname = "nnfw",
+    .fw_opened = 0,
+    .model_file = model_path,
+  };
+
+  test_model = g_build_filename (model_path, "add.tflite", NULL);
+  ASSERT_TRUE (g_file_test (test_model, G_FILE_TEST_EXISTS));
+  g_free (test_model);
+
+  test_model = g_build_filename (model_path, "metadata", "MANIFEST", NULL);
+  ASSERT_TRUE (g_file_test (test_model, G_FILE_TEST_EXISTS));
+  g_free (test_model);
+
+  const GstTensorFilterFramework *sp = nnstreamer_filter_find ("nnfw");
+  EXPECT_NE (sp, (void *) NULL);
+
+  /** get input/output dimension without open */
+  ret = sp->getInputDimension (&prop, &data, &res);
+  EXPECT_NE (ret, 0);
+  ret = sp->getOutputDimension (&prop, &data, &res);
+  EXPECT_NE (ret, 0);
+
+  ret = sp->open (&prop, &data);
+  EXPECT_EQ (ret, 0);
+
+  info.num_tensors = 1;
+  info.info[0].type = _NNS_FLOAT32;
+  info.info[0].dimension[0] = 1;
+  info.info[0].dimension[1] = 1;
+  info.info[0].dimension[2] = 1;
+  info.info[0].dimension[3] = 1;
+
+  /** get input/output dimension successfully */
+  ret = sp->getInputDimension (&prop, &data, NULL);
+  EXPECT_NE (ret, 0);
+  ret = sp->getInputDimension (&prop, &data, &res);
+  EXPECT_EQ (ret, 0);
+
+  EXPECT_EQ (res.num_tensors, info.num_tensors);
+  EXPECT_EQ (res.info[0].type, info.info[0].type);
+  EXPECT_EQ (res.info[0].dimension[0], info.info[0].dimension[0]);
+  EXPECT_EQ (res.info[0].dimension[1], info.info[0].dimension[1]);
+  EXPECT_EQ (res.info[0].dimension[2], info.info[0].dimension[2]);
+  EXPECT_EQ (res.info[0].dimension[3], info.info[0].dimension[3]);
+
+  ret = sp->getOutputDimension (&prop, &data, NULL);
+  EXPECT_NE (ret, 0);
+  ret = sp->getOutputDimension (&prop, &data, &res);
+  EXPECT_EQ (ret, 0);
+
+  EXPECT_EQ (res.num_tensors, info.num_tensors);
+  EXPECT_EQ (res.info[0].type, info.info[0].type);
+  EXPECT_EQ (res.info[0].dimension[0], info.info[0].dimension[0]);
+  EXPECT_EQ (res.info[0].dimension[1], info.info[0].dimension[1]);
+  EXPECT_EQ (res.info[0].dimension[2], info.info[0].dimension[2]);
+  EXPECT_EQ (res.info[0].dimension[3], info.info[0].dimension[3]);
+
+  sp->close (&prop, &data);
+}
+
+/**
+ * @brief Test nnfw subplugin with successful invoke
+ */
+TEST (nnstreamer_nnfw_runtime_raw_functions, invoke)
+{
+  int ret;
+  void *data = NULL;
+  gchar *test_model;
+  gchar *model_path;
+  const gchar *root_path = g_getenv ("NNSTREAMER_BUILD_ROOT_PATH");
+  GstTensorMemory input, output;
+
+  ASSERT_NE (root_path, nullptr);
+
+  /** nnfw needs a directory with model file and metadata in that directory */
+  model_path = g_build_filename (root_path, "tests", "test_models", "models",
+      NULL);
+  GstTensorFilterProperties prop = {
+    .fwname = "nnfw",
+    .fw_opened = 0,
+    .model_file = model_path,
+  };
+
+  /** this model adds 2 to the input data passed in float format */
+  test_model = g_build_filename (model_path, "add.tflite", NULL);
+  ASSERT_TRUE (g_file_test (test_model, G_FILE_TEST_EXISTS));
+  g_free (test_model);
+
+  test_model = g_build_filename (model_path, "metadata", "MANIFEST", NULL);
+  ASSERT_TRUE (g_file_test (test_model, G_FILE_TEST_EXISTS));
+  g_free (test_model);
+
+  const GstTensorFilterFramework *sp = nnstreamer_filter_find ("nnfw");
+  EXPECT_NE (sp, (void *) NULL);
+
+  output.type = input.type = _NNS_FLOAT32;
+  output.size = input.size = sizeof(float) * 1;
+
+  input.data = g_malloc(input.size);
+  output.data = g_malloc(output.size);
+
+  /** invoke without open */
+  ret = sp->invoke_NN (&prop, &data, &input, &output);
+  EXPECT_NE (ret, 0);
+
+  ret = sp->open (&prop, &data);
+  EXPECT_EQ (ret, 0);
+
+  /** invoke successful */
+  ret = sp->invoke_NN (&prop, &data, NULL, NULL);
+  EXPECT_NE (ret, 0);
+  ret = sp->invoke_NN (&prop, &data, &input, NULL);
+  EXPECT_NE (ret, 0);
+  ret = sp->invoke_NN (&prop, &data, NULL, &output);
+  EXPECT_NE (ret, 0);
+
+  *((float *)input.data) = 10.0;
+  ret = sp->invoke_NN (&prop, &data, &input, &output);
+  EXPECT_EQ (ret, 0);
+  EXPECT_EQ (*((float *)output.data), 12.0);
+
+  *((float *)input.data) = 1.0;
+  ret = sp->invoke_NN (&prop, &data, &input, &output);
+  EXPECT_EQ (ret, 0);
+  EXPECT_EQ (*((float *)output.data), 3.0);
+
+  sp->close (&prop, &data);
+}
+
+/**
  * @brief Main gtest
  */
 int