[UCT][nntrainer][ACR-1644] UTC for added function related with dataset

UTC for added function
 - ml_train_dataset_create
 - ml_train_dataset_add_generator
 - ml_train_dataset_add_file
 - ml_train_dataset_set_property_for_mode
and enum ml_train_dataset_mode_e

Change-Id: I8301ad40151c66343e37f2767cb55470098b9179
Signed-off-by: hyeonseok lee <hs89.lee@samsung.com>

diff --git a/src/utc/nntrainer/tct-nntrainer-core_mobile.h b/src/utc/nntrainer/tct-nntrainer-core_mobile.h
index f3752bae74defb77b008a54200252473fe1e702b..721edc826b0529f51b878d509dca2638a683ea34 100644 (file)
--- a/src/utc/nntrainer/tct-nntrainer-core_mobile.h
+++ b/src/utc/nntrainer/tct-nntrainer-core_mobile.h
@@ -130,6 +130,24 @@ extern int utc_nntrainer_model_get_output_tensors_info_p(void);
 extern int utc_nntrainer_model_get_output_tensors_info_n1(void);
 extern int utc_nntrainer_model_get_output_tensors_info_n2(void);
 extern int utc_nntrainer_model_get_output_tensors_info_n3(void);
+extern int utc_nntrainer_dataset_create_p(void);
+extern int utc_nntrainer_dataset_create_n(void);
+extern int utc_nntrainer_dataset_add_generator_p(void);
+extern int utc_nntrainer_dataset_add_generator_train_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_generator_train_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_generator_valid_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_generator_valid_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_generator_test_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_generator_test_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_file_p(void);
+extern int utc_nntrainer_dataset_add_file_train_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_file_train_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_file_valid_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_file_valid_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_file_test_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_file_test_dataset_n2(void);
+extern int utc_nntrainer_dataset_set_property_for_mode_p(void);
+extern int utc_nntrainer_dataset_set_property_for_mode_n(void);
 
 testcase tc_array[] = {
     // layer tcs
@@ -233,6 +251,24 @@ testcase tc_array[] = {
     {"utc_nntrainer_model_get_output_tensors_info_n1", utc_nntrainer_model_get_output_tensors_info_n1, utc_nntrainer_model_startup, utc_nntrainer_model_cleanup},
     {"utc_nntrainer_model_get_output_tensors_info_n2", utc_nntrainer_model_get_output_tensors_info_n2, utc_nntrainer_model_startup, utc_nntrainer_model_cleanup},
     {"utc_nntrainer_model_get_output_tensors_info_n3", utc_nntrainer_model_get_output_tensors_info_n3, utc_nntrainer_model_startup, utc_nntrainer_model_cleanup},
+    {"utc_nntrainer_dataset_create_p", utc_nntrainer_dataset_create_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_create_n", utc_nntrainer_dataset_create_n, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_p", utc_nntrainer_dataset_add_generator_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_train_dataset_n1", utc_nntrainer_dataset_add_generator_train_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_train_dataset_n2", utc_nntrainer_dataset_add_generator_train_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_valid_dataset_n1", utc_nntrainer_dataset_add_generator_valid_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_valid_dataset_n2", utc_nntrainer_dataset_add_generator_valid_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_test_dataset_n1", utc_nntrainer_dataset_add_generator_test_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_test_dataset_n2", utc_nntrainer_dataset_add_generator_test_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_p", utc_nntrainer_dataset_add_file_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_train_dataset_n1", utc_nntrainer_dataset_add_file_train_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_train_dataset_n2", utc_nntrainer_dataset_add_file_train_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_valid_dataset_n1", utc_nntrainer_dataset_add_file_valid_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_valid_dataset_n2", utc_nntrainer_dataset_add_file_valid_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_test_dataset_n1", utc_nntrainer_dataset_add_file_test_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_test_dataset_n2", utc_nntrainer_dataset_add_file_test_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_set_property_for_mode_p", utc_nntrainer_dataset_set_property_for_mode_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_set_property_for_mode_n", utc_nntrainer_dataset_set_property_for_mode_n, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
 
     {NULL, NULL},
 };

diff --git a/src/utc/nntrainer/tct-nntrainer-core_tizeniot.h b/src/utc/nntrainer/tct-nntrainer-core_tizeniot.h
index 132cb2e50d98afda8424c24fe1138e06cfb17a31..e2786d3eb5158e7fe13b77daacf4588dbf31cb83 100644 (file)
--- a/src/utc/nntrainer/tct-nntrainer-core_tizeniot.h
+++ b/src/utc/nntrainer/tct-nntrainer-core_tizeniot.h
@@ -131,6 +131,24 @@ extern int utc_nntrainer_model_get_output_tensors_info_p(void);
 extern int utc_nntrainer_model_get_output_tensors_info_n1(void);
 extern int utc_nntrainer_model_get_output_tensors_info_n2(void);
 extern int utc_nntrainer_model_get_output_tensors_info_n3(void);
+extern int utc_nntrainer_dataset_create_p(void);
+extern int utc_nntrainer_dataset_create_n(void);
+extern int utc_nntrainer_dataset_add_generator_p(void);
+extern int utc_nntrainer_dataset_add_generator_train_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_generator_train_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_generator_valid_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_generator_valid_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_generator_test_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_generator_test_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_file_p(void);
+extern int utc_nntrainer_dataset_add_file_train_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_file_train_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_file_valid_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_file_valid_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_file_test_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_file_test_dataset_n2(void);
+extern int utc_nntrainer_dataset_set_property_for_mode_p(void);
+extern int utc_nntrainer_dataset_set_property_for_mode_n(void);
 
 testcase tc_array[] = {
     // layer tcs
@@ -234,6 +252,24 @@ testcase tc_array[] = {
     {"utc_nntrainer_model_get_output_tensors_info_n1", utc_nntrainer_model_get_output_tensors_info_n1, utc_nntrainer_model_startup, utc_nntrainer_model_cleanup},
     {"utc_nntrainer_model_get_output_tensors_info_n2", utc_nntrainer_model_get_output_tensors_info_n2, utc_nntrainer_model_startup, utc_nntrainer_model_cleanup},
     {"utc_nntrainer_model_get_output_tensors_info_n3", utc_nntrainer_model_get_output_tensors_info_n3, utc_nntrainer_model_startup, utc_nntrainer_model_cleanup},
+    {"utc_nntrainer_dataset_create_p", utc_nntrainer_dataset_create_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_create_n", utc_nntrainer_dataset_create_n, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_p", utc_nntrainer_dataset_add_generator_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_train_dataset_n1", utc_nntrainer_dataset_add_generator_train_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_train_dataset_n2", utc_nntrainer_dataset_add_generator_train_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_valid_dataset_n1", utc_nntrainer_dataset_add_generator_valid_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_valid_dataset_n2", utc_nntrainer_dataset_add_generator_valid_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_test_dataset_n1", utc_nntrainer_dataset_add_generator_test_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_test_dataset_n2", utc_nntrainer_dataset_add_generator_test_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_p", utc_nntrainer_dataset_add_file_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_train_dataset_n1", utc_nntrainer_dataset_add_file_train_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_train_dataset_n2", utc_nntrainer_dataset_add_file_train_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_valid_dataset_n1", utc_nntrainer_dataset_add_file_valid_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_valid_dataset_n2", utc_nntrainer_dataset_add_file_valid_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_test_dataset_n1", utc_nntrainer_dataset_add_file_test_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_test_dataset_n2", utc_nntrainer_dataset_add_file_test_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_set_property_for_mode_p", utc_nntrainer_dataset_set_property_for_mode_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_set_property_for_mode_n", utc_nntrainer_dataset_set_property_for_mode_n, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
 
     {NULL, NULL},
 };

diff --git a/src/utc/nntrainer/tct-nntrainer-core_tv.h b/src/utc/nntrainer/tct-nntrainer-core_tv.h
index 19e0d4827108b90d3f5db6f212ea7741c3b77617..78ceeb9e66a1301bb6afc12f6e2bbed9085a586d 100644 (file)
--- a/src/utc/nntrainer/tct-nntrainer-core_tv.h
+++ b/src/utc/nntrainer/tct-nntrainer-core_tv.h
@@ -131,6 +131,24 @@ extern int utc_nntrainer_model_get_output_tensors_info_p(void);
 extern int utc_nntrainer_model_get_output_tensors_info_n1(void);
 extern int utc_nntrainer_model_get_output_tensors_info_n2(void);
 extern int utc_nntrainer_model_get_output_tensors_info_n3(void);
+extern int utc_nntrainer_dataset_create_p(void);
+extern int utc_nntrainer_dataset_create_n(void);
+extern int utc_nntrainer_dataset_add_generator_p(void);
+extern int utc_nntrainer_dataset_add_generator_train_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_generator_train_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_generator_valid_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_generator_valid_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_generator_test_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_generator_test_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_file_p(void);
+extern int utc_nntrainer_dataset_add_file_train_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_file_train_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_file_valid_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_file_valid_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_file_test_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_file_test_dataset_n2(void);
+extern int utc_nntrainer_dataset_set_property_for_mode_p(void);
+extern int utc_nntrainer_dataset_set_property_for_mode_n(void);
 
 testcase tc_array[] = {
     // layer tcs
@@ -234,6 +252,24 @@ testcase tc_array[] = {
     {"utc_nntrainer_model_get_output_tensors_info_n1", utc_nntrainer_model_get_output_tensors_info_n1, utc_nntrainer_model_startup, utc_nntrainer_model_cleanup},
     {"utc_nntrainer_model_get_output_tensors_info_n2", utc_nntrainer_model_get_output_tensors_info_n2, utc_nntrainer_model_startup, utc_nntrainer_model_cleanup},
     {"utc_nntrainer_model_get_output_tensors_info_n3", utc_nntrainer_model_get_output_tensors_info_n3, utc_nntrainer_model_startup, utc_nntrainer_model_cleanup},
+    {"utc_nntrainer_dataset_create_p", utc_nntrainer_dataset_create_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_create_n", utc_nntrainer_dataset_create_n, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_p", utc_nntrainer_dataset_add_generator_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_train_dataset_n1", utc_nntrainer_dataset_add_generator_train_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_train_dataset_n2", utc_nntrainer_dataset_add_generator_train_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_valid_dataset_n1", utc_nntrainer_dataset_add_generator_valid_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_valid_dataset_n2", utc_nntrainer_dataset_add_generator_valid_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_test_dataset_n1", utc_nntrainer_dataset_add_generator_test_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_test_dataset_n2", utc_nntrainer_dataset_add_generator_test_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_p", utc_nntrainer_dataset_add_file_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_train_dataset_n1", utc_nntrainer_dataset_add_file_train_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_train_dataset_n2", utc_nntrainer_dataset_add_file_train_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_valid_dataset_n1", utc_nntrainer_dataset_add_file_valid_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_valid_dataset_n2", utc_nntrainer_dataset_add_file_valid_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_test_dataset_n1", utc_nntrainer_dataset_add_file_test_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_test_dataset_n2", utc_nntrainer_dataset_add_file_test_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_set_property_for_mode_p", utc_nntrainer_dataset_set_property_for_mode_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_set_property_for_mode_n", utc_nntrainer_dataset_set_property_for_mode_n, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
 
     {NULL, NULL},
 };

diff --git a/src/utc/nntrainer/tct-nntrainer-core_wearable.h b/src/utc/nntrainer/tct-nntrainer-core_wearable.h
index dd0a8a0203fbd0494892746514bd487db09bd6f1..10f47314143f1f97b169954094773aba5bfa3fdb 100644 (file)
--- a/src/utc/nntrainer/tct-nntrainer-core_wearable.h
+++ b/src/utc/nntrainer/tct-nntrainer-core_wearable.h
@@ -131,6 +131,24 @@ extern int utc_nntrainer_model_get_output_tensors_info_p(void);
 extern int utc_nntrainer_model_get_output_tensors_info_n1(void);
 extern int utc_nntrainer_model_get_output_tensors_info_n2(void);
 extern int utc_nntrainer_model_get_output_tensors_info_n3(void);
+extern int utc_nntrainer_dataset_create_p(void);
+extern int utc_nntrainer_dataset_create_n(void);
+extern int utc_nntrainer_dataset_add_generator_p(void);
+extern int utc_nntrainer_dataset_add_generator_train_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_generator_train_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_generator_valid_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_generator_valid_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_generator_test_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_generator_test_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_file_p(void);
+extern int utc_nntrainer_dataset_add_file_train_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_file_train_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_file_valid_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_file_valid_dataset_n2(void);
+extern int utc_nntrainer_dataset_add_file_test_dataset_n1(void);
+extern int utc_nntrainer_dataset_add_file_test_dataset_n2(void);
+extern int utc_nntrainer_dataset_set_property_for_mode_p(void);
+extern int utc_nntrainer_dataset_set_property_for_mode_n(void);
 
 testcase tc_array[] = {
     // layer tcs
@@ -234,6 +252,24 @@ testcase tc_array[] = {
     {"utc_nntrainer_model_get_output_tensors_info_n1", utc_nntrainer_model_get_output_tensors_info_n1, utc_nntrainer_model_startup, utc_nntrainer_model_cleanup},
     {"utc_nntrainer_model_get_output_tensors_info_n2", utc_nntrainer_model_get_output_tensors_info_n2, utc_nntrainer_model_startup, utc_nntrainer_model_cleanup},
     {"utc_nntrainer_model_get_output_tensors_info_n3", utc_nntrainer_model_get_output_tensors_info_n3, utc_nntrainer_model_startup, utc_nntrainer_model_cleanup},
+    {"utc_nntrainer_dataset_create_p", utc_nntrainer_dataset_create_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_create_n", utc_nntrainer_dataset_create_n, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_p", utc_nntrainer_dataset_add_generator_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_train_dataset_n1", utc_nntrainer_dataset_add_generator_train_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_train_dataset_n2", utc_nntrainer_dataset_add_generator_train_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_valid_dataset_n1", utc_nntrainer_dataset_add_generator_valid_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_valid_dataset_n2", utc_nntrainer_dataset_add_generator_valid_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_test_dataset_n1", utc_nntrainer_dataset_add_generator_test_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_generator_test_dataset_n2", utc_nntrainer_dataset_add_generator_test_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_p", utc_nntrainer_dataset_add_file_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_train_dataset_n1", utc_nntrainer_dataset_add_file_train_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_train_dataset_n2", utc_nntrainer_dataset_add_file_train_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_valid_dataset_n1", utc_nntrainer_dataset_add_file_valid_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_valid_dataset_n2", utc_nntrainer_dataset_add_file_valid_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_test_dataset_n1", utc_nntrainer_dataset_add_file_test_dataset_n1, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_add_file_test_dataset_n2", utc_nntrainer_dataset_add_file_test_dataset_n2, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_set_property_for_mode_p", utc_nntrainer_dataset_set_property_for_mode_p, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
+    {"utc_nntrainer_dataset_set_property_for_mode_n", utc_nntrainer_dataset_set_property_for_mode_n, utc_nntrainer_dataset_startup, utc_nntrainer_dataset_cleanup},
 
     {NULL, NULL},
 };

diff --git a/src/utc/nntrainer/utc-nntrainer-dataset.c b/src/utc/nntrainer/utc-nntrainer-dataset.c
index 86bf22f46938393703c94e20cb646fbdd636f965..9678d31c91d8abc63c840143f5c9ee062247b019 100644 (file)
--- a/src/utc/nntrainer/utc-nntrainer-dataset.c
+++ b/src/utc/nntrainer/utc-nntrainer-dataset.c
@@ -232,3 +232,438 @@ int utc_nntrainer_dataset_set_property_n(void)
   ml_train_dataset_destroy(dataset);
   return 0;
 }
+
+/**
+ * @function utc_nntrainer_dataset_create_p
+ * @since_tizen 6.5
+ * @description Test for successfully create dataset
+ */
+int utc_nntrainer_dataset_create_p(void)
+{
+  status = ml_train_dataset_create(&dataset);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_NONE);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_create_n
+ * @since_tizen 6.5
+ * @description Test for fail to create dataset
+ */
+int utc_nntrainer_dataset_create_n(void)
+{
+  status = ml_train_dataset_create(NULL);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_generator_p
+ * @since_tizen 6.5
+ * @description Test for adding generator to dataset
+ */
+int utc_nntrainer_dataset_add_generator_p(void)
+{
+  ml_train_dataset_create(&dataset);
+
+  ml_train_dataset_mode_e mode[3] = {
+    ML_TRAIN_DATASET_MODE_TRAIN, ML_TRAIN_DATASET_MODE_VALID, ML_TRAIN_DATASET_MODE_TEST};
+
+  for (unsigned int i = 0; i < 3; ++i) {
+    status = ml_train_dataset_add_generator(dataset, mode[i], constant_generator_cb, NULL);
+    if (train_supported == false)
+    {
+      assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+    }
+    else
+    {
+      assert_eq(status, ML_ERROR_NONE);
+    }
+  }
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_generator_train_dataset_n1
+ * @since_tizen 6.5
+ * @description Test for adding train dataset generator to NULL dataset
+ */
+int utc_nntrainer_dataset_add_generator_train_dataset_n1(void)
+{
+  status = ml_train_dataset_add_generator(NULL, ML_TRAIN_DATASET_MODE_TRAIN,
+                                          constant_generator_cb, NULL);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_generator_train_dataset_n2
+ * @since_tizen 6.5
+ * @description Test for adding NULL train dataset generator
+ */
+int utc_nntrainer_dataset_add_generator_train_dataset_n2(void)
+{
+  ml_train_dataset_create(&dataset);
+
+  status = ml_train_dataset_add_generator(dataset, ML_TRAIN_DATASET_MODE_TRAIN,
+                                          NULL, NULL);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_generator_valid_dataset_n1
+ * @since_tizen 6.5
+ * @description Test for adding validation dataset generator to NULL dataset
+ */
+int utc_nntrainer_dataset_add_generator_valid_dataset_n1(void)
+{
+  status = ml_train_dataset_add_generator(NULL, ML_TRAIN_DATASET_MODE_VALID,
+                                          constant_generator_cb, NULL);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_generator_valid_dataset_n2
+ * @since_tizen 6.5
+ * @description Test for adding NULL validation dataset generator
+ */
+int utc_nntrainer_dataset_add_generator_valid_dataset_n2(void)
+{
+  ml_train_dataset_create(&dataset);
+
+  status = ml_train_dataset_add_generator(dataset, ML_TRAIN_DATASET_MODE_VALID,
+                                          NULL, NULL);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_generator_test_dataset_n1
+ * @since_tizen 6.5
+ * @description Test for adding test dataset generator to NULL dataset
+ */
+int utc_nntrainer_dataset_add_generator_test_dataset_n1(void)
+{
+  status = ml_train_dataset_add_generator(NULL, ML_TRAIN_DATASET_MODE_TEST,
+                                          constant_generator_cb, NULL);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_generator_test_dataset_n2
+ * @since_tizen 6.5
+ * @description Test for adding NULL test dataset generator
+ */
+int utc_nntrainer_dataset_add_generator_test_dataset_n2(void)
+{
+  ml_train_dataset_create(&dataset);
+
+  status = ml_train_dataset_add_generator(dataset, ML_TRAIN_DATASET_MODE_TEST,
+                                          NULL, NULL);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_file_p
+ * @since_tizen 6.5
+ * @description Test for adding data file to dataset
+ */
+int utc_nntrainer_dataset_add_file_p(void)
+{
+  ml_train_dataset_create(&dataset);
+
+  ml_train_dataset_mode_e mode[3] = {
+    ML_TRAIN_DATASET_MODE_TRAIN, ML_TRAIN_DATASET_MODE_VALID, ML_TRAIN_DATASET_MODE_TEST};
+
+  for (unsigned int i = 0; i < 3; ++i) {
+    status = ml_train_dataset_add_file(dataset, mode[i], train_path);
+    if (train_supported == false)
+    {
+      assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+    }
+    else
+    {
+      assert_eq(status, ML_ERROR_NONE);
+    }
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_file_train_dataset_n1
+ * @since_tizen 6.5
+ * @description Test for adding train dataset data file to NULL dataset
+ */
+int utc_nntrainer_dataset_add_file_train_dataset_n1(void)
+{
+  status = ml_train_dataset_add_file(NULL, ML_TRAIN_DATASET_MODE_TRAIN, train_path);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_file_train_dataset_n2
+ * @since_tizen 6.5
+ * @description Test for adding train dataset data file with NULL file path
+ */
+int utc_nntrainer_dataset_add_file_train_dataset_n2(void)
+{
+  ml_train_dataset_create(&dataset);
+
+  status = ml_train_dataset_add_file(dataset, ML_TRAIN_DATASET_MODE_TRAIN, NULL);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_file_valid_dataset_n1
+ * @since_tizen 6.5
+ * @description Test for adding validation dataset data file to NULL dataset
+ */
+int utc_nntrainer_dataset_add_file_valid_dataset_n1(void)
+{
+  status = ml_train_dataset_add_file(NULL, ML_TRAIN_DATASET_MODE_VALID, train_path);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_file_valid_dataset_n2
+ * @since_tizen 6.5
+ * @description Test for adding validation dataset data file with NULL file path
+ */
+int utc_nntrainer_dataset_add_file_valid_dataset_n2(void)
+{
+  ml_train_dataset_create(&dataset);
+
+  status = ml_train_dataset_add_file(dataset, ML_TRAIN_DATASET_MODE_VALID, NULL);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_file_test_dataset_n1
+ * @since_tizen 6.5
+ * @description Test for adding test dataset data file to NULL dataset
+ */
+int utc_nntrainer_dataset_add_file_test_dataset_n1(void)
+{
+  status = ml_train_dataset_add_file(NULL, ML_TRAIN_DATASET_MODE_TEST, train_path);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_add_file_test_dataset_n2
+ * @since_tizen 6.5
+ * @description Test for adding test dataset data file with NULL file path
+ */
+int utc_nntrainer_dataset_add_file_test_dataset_n2(void)
+{
+  ml_train_dataset_create(&dataset);
+
+  status = ml_train_dataset_add_file(dataset, ML_TRAIN_DATASET_MODE_TEST, NULL);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_set_property_for_mode_p
+ * @since_tizen 6.5
+ * @description Test for dataset set property for mode
+ */
+int utc_nntrainer_dataset_set_property_for_mode_p(void)
+{
+  ml_train_dataset_create(&dataset);
+
+  ml_train_dataset_add_generator(dataset, ML_TRAIN_DATASET_MODE_TRAIN, constant_generator_cb, NULL);
+
+  status = ml_train_dataset_set_property_for_mode(
+    dataset, ML_TRAIN_DATASET_MODE_TRAIN, "buffer_size=1", NULL);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_NONE);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}
+
+/**
+ * @function utc_nntrainer_dataset_set_property_for_mode_n
+ * @since_tizen 6.5
+ * @description Test for dataset set property for mode with empty mode
+ */
+int utc_nntrainer_dataset_set_property_for_mode_n(void)
+{
+  ml_train_dataset_create(&dataset);
+
+  status = ml_train_dataset_set_property_for_mode(
+    dataset, ML_TRAIN_DATASET_MODE_TRAIN, "buffer_size=1", NULL);
+  if (train_supported == false)
+  {
+    assert_eq(status, ML_ERROR_NOT_SUPPORTED);
+  }
+  else
+  {
+    assert_eq(status, ML_ERROR_INVALID_PARAMETER);
+  }
+
+  ml_train_dataset_destroy(dataset);
+
+  return 0;
+}