mmi-vision-provider-tests : add test about mv_engine_config_h, and mv_inference_task_...

Change-Id: I4fd793209a2eb0226d133390739fdeb1d404ca3c

diff --git a/tests/vision-provider/mmi-vision-provider-tests.cpp b/tests/vision-provider/mmi-vision-provider-tests.cpp
index 425d67a..3ad0914 100644 (file)
--- a/tests/vision-provider/mmi-vision-provider-tests.cpp
+++ b/tests/vision-provider/mmi-vision-provider-tests.cpp
@@ -82,16 +82,59 @@ public:
        }
 };
 
-TEST_F(MMIVisionProviderVisionTest, MMIVisionProviderVisionInit)
+#define IC_LABEL_MMI_TEACHABLE_PATH\
+       "/usr/share/mmi-provider/labels.txt"
+#define IC_TFLITE_WEIGHT_MMI_TEACHABLE_PATH \
+       "/usr/share/mmi-provider/model_quant.tflite"
+#define IC_TFLITE_META_MMI_TEACHABLE_PATH \
+       "/usr/share/mmi-provider/model_quant.json"
+
+#define TASK_IC 0
+
+TEST_F(MMIVisionProviderVisionTest, MMIVisionProviderMVEngine)
 {
-       int res = 0;
-       res = vision_init();
+       int res = MEDIA_VISION_ERROR_NONE;
+       mv_engine_config_h engine_cfg = NULL;
 
+       res = mv_create_engine_config(&engine_cfg);
        EXPECT_EQ(res, 0);
+       EXPECT_NE(engine_cfg, nullptr);
 
-       res = vision_shutdown();
+       res = engine_config_user_hosted_tflite_cpu(
+                       engine_cfg, IC_TFLITE_WEIGHT_MMI_TEACHABLE_PATH,
+                       IC_LABEL_MMI_TEACHABLE_PATH,
+                       IC_TFLITE_META_MMI_TEACHABLE_PATH);
 
        EXPECT_EQ(res, 0);
+       EXPECT_NE(engine_cfg, nullptr);
+       if(engine_cfg)
+       {
+               mv_destroy_engine_config(engine_cfg);
+               engine_cfg = NULL;
+       }
+       vision_shutdown();
+}
+
+TEST_F(MMIVisionProviderVisionTest, MMIVisionProviderMVTaskHelper)
+{
+       int res = MEDIA_VISION_ERROR_NONE;
+       mv_engine_config_h engine_cfg = NULL;
+
+       res = mv_create_engine_config(&engine_cfg);
+
+       res = engine_config_user_hosted_tflite_cpu(
+                       engine_cfg, IC_TFLITE_WEIGHT_MMI_TEACHABLE_PATH,
+                       IC_LABEL_MMI_TEACHABLE_PATH,
+                       IC_TFLITE_META_MMI_TEACHABLE_PATH);
+
+       res = mv_inference_task_helper(engine_cfg, TASK_IC);
+       EXPECT_EQ(res, 0);
+       EXPECT_NE(engine_cfg, nullptr);
+       if(engine_cfg)
+       {
+               mv_destroy_engine_config(engine_cfg);
+       }
+       vision_shutdown();
 }
 
 TEST_F(MMIVisionProviderVisionTest, MMIVisionProviderClassificationInit)
@@ -105,6 +148,18 @@ TEST_F(MMIVisionProviderVisionTest, MMIVisionProviderClassificationInit)
        vision_shutdown();
 }
 
+TEST_F(MMIVisionProviderVisionTest, MMIVisionProviderVisionInit)
+{
+       int res = 0;
+       res = vision_init();
+
+       EXPECT_EQ(res, 0);
+
+       res = vision_shutdown();
+
+       EXPECT_EQ(res, 0);
+}
+
 static Eina_Bool
 _vison_video_working_finish(void *data, int type, void *event)
 {