[TEST] Add apply option testcases for tensor transform

Add apply option testcases for tensor transform

Signed-off-by: Gichan Jang <gichan2.jang@samsung.com>

diff --git a/tests/nnstreamer_sink/unittest_sink.cc b/tests/nnstreamer_sink/unittest_sink.cc
index 859b779..ecdbfb5 100644 (file)
--- a/tests/nnstreamer_sink/unittest_sink.cc
+++ b/tests/nnstreamer_sink/unittest_sink.cc
@@ -140,6 +140,8 @@ typedef enum {
                                       tensor_transform (arithmetic mode) */
   TEST_TYPE_TRANSFORM_TENSORS,  /**< pipeline for tensors
                                      with tensor_transform (typecast mode) */
+  TEST_TYPE_TRANSFORM_APPLY,  /**< pipeline for apply option
+                                     with tensor_transform (typecast mode) */
   TEST_TYPE_TYPECAST, /**< pipeline for typecast with tensor_transform */
   TEST_TYPE_ISSUE739_MUX_PARALLEL_1, /**< pipeline to test Mux/Parallel case in
                                         #739 */
@@ -926,6 +928,15 @@ _setup_pipeline (TestOption &option)
         "videotestsrc num-buffers=%d ! videoconvert ! video/x-raw,width=320,height=240,format=RGB,framerate=(fraction)30/1 ! tensor_converter ! mux.sink_2",
         option.num_buffers, option.num_buffers, option.num_buffers);
     break;
+  case TEST_TYPE_TRANSFORM_APPLY:
+    /* tensors stream with apply option, typecast to float32 */
+    str_pipeline = g_strdup_printf (
+        "tensor_mux name=mux ! tensor_transform mode=typecast option=float32 apply=2,0 ! tensor_sink name=test_sink "
+        "videotestsrc num-buffers=%d ! videoconvert ! video/x-raw,width=160,height=120,format=RGB,framerate=(fraction)30/1 ! tensor_converter ! mux.sink_0 "
+        "videotestsrc num-buffers=%d ! videoconvert ! video/x-raw,width=280,height=40,format=RGB,framerate=(fraction)30/1 ! tensor_converter ! mux.sink_1 "
+        "videotestsrc num-buffers=%d ! videoconvert ! video/x-raw,width=320,height=240,format=RGB,framerate=(fraction)30/1 ! tensor_converter ! mux.sink_2",
+        option.num_buffers, option.num_buffers, option.num_buffers);
+    break;
   case TEST_TYPE_TYPECAST:
     /** text stream to test typecast */
     str_pipeline = g_strdup_printf (
@@ -4471,6 +4482,66 @@ TEST (tensorStreamTest, transformTensors)
 }
 
 /**
+ * @brief Test for apply option of the tensor transform
+ */
+TEST (tensorStreamTest, transformApplyOpt)
+{
+  const guint num_buffers = 5;
+  TestOption option = { num_buffers, TEST_TYPE_TRANSFORM_APPLY };
+
+  ASSERT_TRUE (_setup_pipeline (option));
+
+  gst_element_set_state (g_test_data.pipeline, GST_STATE_PLAYING);
+  g_main_loop_run (g_test_data.loop);
+
+  EXPECT_TRUE (_wait_pipeline_process_buffers (num_buffers));
+  gst_element_set_state (g_test_data.pipeline, GST_STATE_NULL);
+
+  /** check eos message */
+  EXPECT_EQ (g_test_data.status, TEST_EOS);
+
+  /** check received buffers */
+  EXPECT_EQ (g_test_data.received, num_buffers);
+  EXPECT_EQ (g_test_data.mem_blocks, 3U);
+  EXPECT_EQ (g_test_data.received_size, 1185600U);
+  /** (160 * 120 * 3 + 320 * 240 * 3) * 4 +  (280 * 40 * 3) */
+
+  /** check caps name */
+  EXPECT_STREQ (g_test_data.caps_name, "other/tensors");
+
+  /** check timestamp */
+  EXPECT_FALSE (g_test_data.invalid_timestamp);
+
+  /** check tensors config for video */
+  EXPECT_TRUE (gst_tensors_config_validate (&g_test_data.tensors_config));
+  EXPECT_EQ (g_test_data.tensors_config.info.num_tensors, 3U);
+
+  EXPECT_EQ (g_test_data.tensors_config.info.info[0].type, _NNS_FLOAT32);
+  EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[0], 3U);
+  EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[1], 160U);
+  EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[2], 120U);
+  EXPECT_EQ (g_test_data.tensors_config.info.info[0].dimension[3], 1U);
+
+  EXPECT_EQ (g_test_data.tensors_config.info.info[1].type, _NNS_UINT8);
+  EXPECT_EQ (g_test_data.tensors_config.info.info[1].dimension[0], 3U);
+  EXPECT_EQ (g_test_data.tensors_config.info.info[1].dimension[1], 280U);
+  EXPECT_EQ (g_test_data.tensors_config.info.info[1].dimension[2], 40U);
+  EXPECT_EQ (g_test_data.tensors_config.info.info[1].dimension[3], 1U);
+
+  EXPECT_EQ (g_test_data.tensors_config.info.info[2].type, _NNS_FLOAT32);
+  EXPECT_EQ (g_test_data.tensors_config.info.info[2].dimension[0], 3U);
+  EXPECT_EQ (g_test_data.tensors_config.info.info[2].dimension[1], 320U);
+  EXPECT_EQ (g_test_data.tensors_config.info.info[2].dimension[2], 240U);
+  EXPECT_EQ (g_test_data.tensors_config.info.info[2].dimension[3], 1U);
+
+  EXPECT_EQ (g_test_data.tensors_config.rate_n, 30);
+  EXPECT_EQ (g_test_data.tensors_config.rate_d, 1);
+
+  EXPECT_FALSE (g_test_data.test_failed);
+  _free_test_data (option);
+}
+
+/**
  * @brief Test for video stream with tensor_split.
  */
 TEST (tensorStreamTest, videoSplit)

diff --git a/tests/transform_typecast/runTest.sh b/tests/transform_typecast/runTest.sh
index 6ac7e89..9a65104 100644 (file)
--- a/tests/transform_typecast/runTest.sh
+++ b/tests/transform_typecast/runTest.sh
@@ -91,7 +91,7 @@ gstTest "--gst-plugin-path=${PATH_TO_PLUGIN} multifilesrc location=\"testsequenc
         t. ! queue ! mux.sink_0 \
         t. ! queue ! mux.sink_1 \
         t. ! queue ! mux.sink_2 \
-        tensor_mux name=mux ! tensor_transform mode=typecast option=float32 ! tensor_demux name=demux \
+        tensor_mux name=mux ! tensor_transform mode=typecast option=float32 apply=0,1,2 ! tensor_demux name=demux \
         demux.src_0 ! queue ! filesink location=\"testcase_tensors_0.typecast.log\" buffer-mode=unbuffered sync=false async=false \
         demux.src_1 ! queue ! filesink location=\"testcase_tensors_1.typecast.log\" buffer-mode=unbuffered sync=false async=false \
         demux.src_2 ! queue ! filesink location=\"testcase_tensors_2.typecast.log\" buffer-mode=unbuffered sync=false async=false" 10 0 0 $PERFORMANCE
@@ -102,4 +102,21 @@ testResult $? 10 "Golden test comparison 1" 0 1
 python3 checkResult.py typecast testcase_tensors.direct.log testcase_tensors_2.typecast.log uint8 1 B float32 4 f
 testResult $? 10 "Golden test comparison 2" 0 1
 
+# Tensor tensors stream with apply option
+gstTest "--gst-plugin-path=${PATH_TO_PLUGIN} multifilesrc location=\"testsequence_%1d.png\" index=0 caps=\"image/png,framerate=\(fraction\)30/1\" ! \
+        pngdec ! videoconvert ! video/x-raw, format=BGRx ! tensor_converter ! tee name=t \
+        t. ! queue ! filesink location=\"testcase_apply.direct.log\" buffer-mode=unbuffered sync=false async=false \
+        t. ! queue ! mux.sink_0 \
+        t. ! queue ! mux.sink_1 \
+        t. ! queue ! mux.sink_2 \
+        tensor_mux name=mux ! tensor_transform mode=typecast option=float32 apply=0,2 ! tensor_demux name=demux \
+        demux.src_0 ! queue ! filesink location=\"testcase_apply_0.typecast.log\" buffer-mode=unbuffered sync=false async=false \
+        demux.src_1 ! queue ! filesink location=\"testcase_apply_1.typecast.log\" buffer-mode=unbuffered sync=false async=false \
+        demux.src_2 ! queue ! filesink location=\"testcase_apply_2.typecast.log\" buffer-mode=unbuffered sync=false async=false" 11 0 0 $PERFORMANCE
+python3 checkResult.py typecast testcase_apply.direct.log testcase_apply_0.typecast.log uint8 1 B float32 4 f
+testResult $? 10 "apply test comparison 11-1" 0 1
+callCompareTest testcase_apply.direct.log testcase_apply_1.typecast.log 11-2 "apply test comparison 1-1" 1 0
+python3 checkResult.py typecast testcase_apply.direct.log testcase_apply_2.typecast.log uint8 1 B float32 4 f
+testResult $? 10 "apply test comparison 11-2" 0 1
+
 report