[TEST] Add unittest related to setProperty for dataset

Add unittest for ml_train_dataset_set_property_for_mode_with_single_param()

**Self evaluation:**
1. Build test:   [X]Passed [ ]Failed [ ]Skipped
2. Run test:     [X]Passed [ ]Failed [ ]Skipped

Signed-off-by: Hyunil <hyunil46.park@samsung.com>

diff --git a/test/tizen_capi/unittest_tizen_capi_dataset.cpp b/test/tizen_capi/unittest_tizen_capi_dataset.cpp
index 1023d1bd3b5b916f17b9675811f8b3eb3d7fa6aa..aad12b35101254762f281a778c5befa534763e9a 100644 (file)
--- a/test/tizen_capi/unittest_tizen_capi_dataset.cpp
+++ b/test/tizen_capi/unittest_tizen_capi_dataset.cpp
@@ -566,6 +566,61 @@ TEST(nntrainer_capi_dataset, set_dataset_property_for_mode_02_n) {
   EXPECT_EQ(status, ML_ERROR_NONE);
 }
 
+/**
+ * @brief Neural Network Dataset set Property Test (positive test)
+ */
+TEST(nntrainer_capi_dataset,
+     set_dataset_property_for_mode_with_single_param_03_p) {
+  ml_train_dataset_h dataset;
+  int status = ML_ERROR_NONE;
+
+  status = ml_train_dataset_create(&dataset);
+  EXPECT_EQ(status, ML_ERROR_NONE);
+
+  auto train_data = createTrainData();
+  auto valid_data = createValidData();
+  status = ml_train_dataset_add_generator(dataset, ML_TRAIN_DATASET_MODE_TRAIN,
+                                          getSample, &train_data);
+  EXPECT_EQ(status, ML_ERROR_NONE);
+  status = ml_train_dataset_set_property_for_mode_with_single_param(
+    dataset, ML_TRAIN_DATASET_MODE_TRAIN, "buffer_size=1");
+  EXPECT_EQ(status, ML_ERROR_NONE);
+
+  status = ml_train_dataset_add_generator(dataset, ML_TRAIN_DATASET_MODE_VALID,
+                                          getSample, &valid_data);
+  status = ml_train_dataset_set_property_for_mode_with_single_param(
+    dataset, ML_TRAIN_DATASET_MODE_VALID, "buffer_size=1");
+  EXPECT_EQ(status, ML_ERROR_NONE);
+
+  status = ml_train_dataset_add_generator(dataset, ML_TRAIN_DATASET_MODE_TEST,
+                                          getSample, &train_data);
+  status = ml_train_dataset_set_property_for_mode_with_single_param(
+    dataset, ML_TRAIN_DATASET_MODE_TEST, "buffer_size=1");
+  EXPECT_EQ(status, ML_ERROR_NONE);
+
+  status = ml_train_dataset_destroy(dataset);
+  EXPECT_EQ(status, ML_ERROR_NONE);
+}
+
+/**
+ * @brief Neural Network Dataset set Property Test (negative test)
+ */
+TEST(nntrainer_capi_dataset,
+     set_dataset_property_for_mode_with_single_param_04_n) {
+  ml_train_dataset_h dataset;
+  int status = ML_ERROR_NONE;
+
+  status = ml_train_dataset_create(&dataset);
+  EXPECT_EQ(status, ML_ERROR_NONE);
+
+  status = ml_train_dataset_set_property_for_mode_with_single_param(
+    dataset, ML_TRAIN_DATASET_MODE_TRAIN, "buffer_size=1");
+  EXPECT_EQ(status, ML_ERROR_INVALID_PARAMETER);
+
+  status = ml_train_dataset_destroy(dataset);
+  EXPECT_EQ(status, ML_ERROR_NONE);
+}
+
 /**
  * @brief Neural Network Dataset set Property Test (negative test)
  */