2020-03-27 14:49:12 +08:00
|
|
|
include(GoogleTest)
|
|
|
|
|
|
|
|
|
|
SET(DE_UT_SRCS
|
2021-02-17 12:48:22 +08:00
|
|
|
affine_op_test.cc
|
2020-11-28 05:52:34 +08:00
|
|
|
execute_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
arena_test.cc
|
|
|
|
|
auto_contrast_op_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
batch_op_test.cc
|
|
|
|
|
bit_functions_test.cc
|
2020-07-07 06:27:32 +08:00
|
|
|
bounding_box_augment_op_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
btree_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
buddy_test.cc
|
|
|
|
|
build_vocab_test.cc
|
2021-08-22 16:26:45 +08:00
|
|
|
c_api_audio_a_to_q_test.cc
|
|
|
|
|
c_api_audio_r_to_z_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_cache_test.cc
|
2021-11-10 15:51:19 +08:00
|
|
|
c_api_dataset_ag_news_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_dataset_album_test.cc
|
2021-12-23 16:32:31 +08:00
|
|
|
c_api_dataset_amazon_review_test.cc
|
2021-12-28 19:04:18 +08:00
|
|
|
c_api_dataset_caltech256_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_dataset_cifar_test.cc
|
2021-08-31 20:03:37 +08:00
|
|
|
c_api_dataset_cityscapes_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_dataset_clue_test.cc
|
2022-01-20 17:40:35 +08:00
|
|
|
c_api_dataset_cmu_arctic_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_dataset_coco_test.cc
|
2021-12-21 15:31:04 +08:00
|
|
|
c_api_dataset_conll2000_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_dataset_config_test.cc
|
|
|
|
|
c_api_dataset_csv_test.cc
|
2021-11-12 11:02:22 +08:00
|
|
|
c_api_dataset_dbpedia_test.cc
|
2021-09-10 09:37:36 +08:00
|
|
|
c_api_dataset_div2k_test.cc
|
2021-08-19 18:23:15 +08:00
|
|
|
c_api_dataset_emnist_test.cc
|
2022-01-04 13:45:43 +08:00
|
|
|
c_api_dataset_en_wik9_test.cc
|
2021-08-19 19:40:55 +08:00
|
|
|
c_api_dataset_fake_image_test.cc
|
2021-09-17 20:52:44 +08:00
|
|
|
c_api_dataset_fashion_mnist_test.cc
|
2021-08-22 16:26:45 +08:00
|
|
|
c_api_dataset_flickr_test.cc
|
2022-01-19 13:48:50 +08:00
|
|
|
c_api_dataset_gtzan_test.cc
|
2021-11-20 13:47:15 +08:00
|
|
|
c_api_dataset_imdb_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_dataset_iterator_test.cc
|
2021-12-13 21:47:55 +08:00
|
|
|
c_api_dataset_iwslt_test.cc
|
2021-12-16 11:38:55 +08:00
|
|
|
c_api_dataset_kitti_test.cc
|
2021-09-14 17:45:38 +08:00
|
|
|
c_api_dataset_kmnist_test.cc
|
2021-12-19 10:28:17 +08:00
|
|
|
c_api_dataset_lfw_test.cc
|
2022-01-21 17:34:42 +08:00
|
|
|
c_api_dataset_libri_tts.cc
|
2021-11-14 14:52:15 +08:00
|
|
|
c_api_dataset_lj_speech_test.cc
|
2021-12-08 16:34:04 +08:00
|
|
|
c_api_dataset_lsun_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_dataset_manifest_test.cc
|
|
|
|
|
c_api_dataset_minddata_test.cc
|
2022-01-19 18:33:13 +08:00
|
|
|
c_api_dataset_multi30k_test.cc
|
2021-12-16 16:03:48 +08:00
|
|
|
c_api_dataset_omniglot_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_dataset_ops_test.cc
|
2021-12-08 17:22:30 +08:00
|
|
|
c_api_dataset_penn_treebank_test.cc
|
2021-10-19 21:58:55 +08:00
|
|
|
c_api_dataset_photo_tour_test.cc
|
2021-09-30 00:41:36 +08:00
|
|
|
c_api_dataset_places365_test.cc
|
2021-07-11 02:15:56 +08:00
|
|
|
c_api_dataset_qmnist_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_dataset_randomdata_test.cc
|
|
|
|
|
c_api_dataset_save.cc
|
2021-06-02 16:52:40 +08:00
|
|
|
c_api_dataset_sbu_test.cc
|
2021-12-27 19:49:06 +08:00
|
|
|
c_api_dataset_semeion_test.cc
|
2021-12-15 14:53:38 +08:00
|
|
|
c_api_dataset_sogou_news_test.cc
|
2021-11-27 11:25:18 +08:00
|
|
|
c_api_dataset_speech_commands_test.cc
|
2022-02-08 18:58:06 +08:00
|
|
|
c_api_dataset_squad_test.cc
|
2021-12-07 09:21:11 +08:00
|
|
|
c_api_dataset_stl10_test.cc
|
2021-12-02 09:53:39 +08:00
|
|
|
c_api_dataset_tedlium_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_dataset_textfile_test.cc
|
|
|
|
|
c_api_dataset_tfrecord_test.cc
|
2021-12-18 23:28:05 +08:00
|
|
|
c_api_dataset_udpos_test.cc
|
2021-06-02 13:56:12 +08:00
|
|
|
c_api_dataset_usps_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_dataset_voc_test.cc
|
2021-12-31 13:49:47 +08:00
|
|
|
c_api_dataset_wider_face_test.cc
|
2021-12-16 14:18:01 +08:00
|
|
|
c_api_dataset_yahoo_answers_test.cc
|
2021-12-07 17:09:41 +08:00
|
|
|
c_api_dataset_yelp_review_test.cc
|
2021-11-19 13:36:24 +08:00
|
|
|
c_api_dataset_yes_no_test.cc
|
2021-12-29 17:24:15 +08:00
|
|
|
c_api_dataset_wiki_text_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_datasets_test.cc
|
2020-12-10 10:50:22 +08:00
|
|
|
c_api_epoch_ctrl_test.cc
|
2021-01-14 07:02:39 +08:00
|
|
|
c_api_pull_based_test.cc
|
2020-12-10 10:50:22 +08:00
|
|
|
c_api_repeat_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_samplers_test.cc
|
|
|
|
|
c_api_text_sentence_piece_vocab_test.cc
|
2021-11-04 20:42:41 +08:00
|
|
|
c_api_text_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_text_vocab_test.cc
|
2021-11-04 20:42:41 +08:00
|
|
|
c_api_text_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
c_api_transforms_test.cc
|
2021-02-23 22:53:26 +08:00
|
|
|
c_api_vision_a_to_q_test.cc
|
2021-03-03 01:14:26 +08:00
|
|
|
c_api_vision_affine_test.cc
|
2021-02-23 22:53:26 +08:00
|
|
|
c_api_vision_bounding_box_augment_test.cc
|
2021-05-25 15:24:41 +08:00
|
|
|
c_api_vision_horizontal_flip_test.cc
|
2021-02-23 22:53:26 +08:00
|
|
|
c_api_vision_random_subselect_policy_test.cc
|
|
|
|
|
c_api_vision_random_test.cc
|
|
|
|
|
c_api_vision_r_to_z_test.cc
|
2021-06-10 19:26:42 +08:00
|
|
|
c_api_vision_slice_patches_test.cc
|
2021-02-23 22:53:26 +08:00
|
|
|
c_api_vision_uniform_aug_test.cc
|
2021-05-25 15:24:41 +08:00
|
|
|
c_api_vision_vertical_flip_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
center_crop_op_test.cc
|
|
|
|
|
channel_swap_test.cc
|
|
|
|
|
circular_pool_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
coco_op_test.cc
|
|
|
|
|
common/bboxop_common.cc
|
|
|
|
|
common/common.cc
|
|
|
|
|
common/cvop_common.cc
|
|
|
|
|
concatenate_op_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
connector_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
csv_op_test.cc
|
2020-06-20 03:00:34 +08:00
|
|
|
cut_out_op_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
cutmix_batch_op_test.cc
|
|
|
|
|
cyclic_array_test.cc
|
|
|
|
|
data_helper_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
datatype_test.cc
|
|
|
|
|
decode_op_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
distributed_sampler_test.cc
|
2020-07-28 02:32:24 +08:00
|
|
|
equalize_op_test.cc
|
2021-12-06 19:24:37 +08:00
|
|
|
execute_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
execution_tree_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
fill_op_test.cc
|
2021-05-17 11:44:43 +08:00
|
|
|
c_api_vision_gaussian_blur_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
global_context_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
gnn_graph_test.cc
|
|
|
|
|
image_process_test.cc
|
|
|
|
|
interrupt_test.cc
|
2020-12-10 10:50:22 +08:00
|
|
|
ir_callback_test.cc
|
2021-03-01 23:32:24 +08:00
|
|
|
ir_sampler_test.cc
|
2020-12-10 10:50:22 +08:00
|
|
|
ir_tensor_op_fusion_pass_test.cc
|
|
|
|
|
ir_tree_adapter_test.cc
|
2021-03-03 01:14:26 +08:00
|
|
|
ir_vision_random_test.cc
|
2021-02-24 10:18:04 +08:00
|
|
|
ir_vision_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
jieba_tokenizer_op_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
main_test.cc
|
|
|
|
|
map_op_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
mask_test.cc
|
|
|
|
|
memory_pool_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
mind_record_op_test.cc
|
2020-07-31 01:24:04 +08:00
|
|
|
mixup_batch_op_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
normalize_op_test.cc
|
|
|
|
|
one_hot_op_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
optimization_pass_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
pad_end_op_test.cc
|
2020-06-20 03:00:34 +08:00
|
|
|
pad_op_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
path_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
perf_data_test.cc
|
2021-10-30 03:16:47 +08:00
|
|
|
profiler_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
queue_test.cc
|
2020-07-28 02:32:24 +08:00
|
|
|
random_affine_op_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
random_color_adjust_op_test.cc
|
2020-08-14 02:30:27 +08:00
|
|
|
random_color_op_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
random_crop_and_resize_op_test.cc
|
2020-07-09 23:32:13 +08:00
|
|
|
random_crop_and_resize_with_bbox_op_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
random_crop_decode_resize_op_test.cc
|
|
|
|
|
random_crop_op_test.cc
|
|
|
|
|
random_crop_with_bbox_op_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
random_horizontal_flip_op_test.cc
|
2020-07-07 06:27:32 +08:00
|
|
|
random_horizontal_flip_with_bbox_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
random_resize_op_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
random_resize_op_test.cc
|
2020-07-09 23:20:19 +08:00
|
|
|
random_resize_with_bbox_op_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
random_rotation_op_test.cc
|
2020-08-11 05:50:01 +08:00
|
|
|
random_solarize_op_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
random_vertical_flip_op_test.cc
|
2020-07-09 23:32:13 +08:00
|
|
|
random_vertical_flip_with_bbox_op_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
rescale_op_test.cc
|
|
|
|
|
resize_op_test.cc
|
2020-07-09 23:20:19 +08:00
|
|
|
resize_with_bbox_op_test.cc
|
2020-08-29 04:54:15 +08:00
|
|
|
rgba_to_bgr_op_test.cc
|
|
|
|
|
rgba_to_rgb_op_test.cc
|
2020-07-27 23:31:00 +08:00
|
|
|
schema_test.cc
|
2022-03-09 00:45:45 +08:00
|
|
|
skip_first_epoch_sampler_test.cc
|
2022-03-09 03:17:18 +08:00
|
|
|
skip_pushdown_optimization_pass_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
slice_op_test.cc
|
|
|
|
|
sliding_window_op_test.cc
|
|
|
|
|
solarize_op_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
stand_alone_samplers_test.cc
|
|
|
|
|
status_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
storage_container_test.cc
|
|
|
|
|
subset_random_sampler_test.cc
|
2021-01-26 01:40:38 +08:00
|
|
|
subset_sampler_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
swap_red_blue_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
task_manager_test.cc
|
2020-09-15 05:34:23 +08:00
|
|
|
tensor_row_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
tensor_string_test.cc
|
2020-10-15 23:09:40 +08:00
|
|
|
tensor_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
tensorshape_test.cc
|
|
|
|
|
tfReader_op_test.cc
|
|
|
|
|
to_float16_op_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
tokenizer_op_test.cc
|
|
|
|
|
treap_test.cc
|
2021-01-09 11:28:43 +08:00
|
|
|
tree_modifying_function_test.cc
|
2020-11-17 05:16:56 +08:00
|
|
|
trucate_pair_test.cc
|
2020-06-19 02:33:08 +08:00
|
|
|
type_cast_op_test.cc
|
|
|
|
|
weighted_random_sampler_test.cc
|
|
|
|
|
)
|
2020-03-27 14:49:12 +08:00
|
|
|
|
2021-01-09 11:28:43 +08:00
|
|
|
if(ENABLE_PYTHON)
|
2020-07-27 23:31:00 +08:00
|
|
|
set(DE_UT_SRCS
|
|
|
|
|
${DE_UT_SRCS}
|
|
|
|
|
manifest_op_test.cc
|
|
|
|
|
)
|
2021-01-09 11:28:43 +08:00
|
|
|
endif()
|
2020-07-27 23:31:00 +08:00
|
|
|
|
2021-02-05 16:46:24 +08:00
|
|
|
if(ENABLE_ACL)
|
|
|
|
|
set(DE_UT_SRCS
|
|
|
|
|
${DE_UT_SRCS}
|
|
|
|
|
dvpp_decode_jpeg_test.cc)
|
|
|
|
|
endif()
|
|
|
|
|
|
2020-03-27 14:49:12 +08:00
|
|
|
add_executable(de_ut_tests ${DE_UT_SRCS})
|
|
|
|
|
|
|
|
|
|
set_target_properties(de_ut_tests PROPERTIES INSTALL_RPATH "$ORIGIN/../lib:$ORIGIN/../lib64")
|
|
|
|
|
|
2021-01-09 11:28:43 +08:00
|
|
|
target_link_libraries(de_ut_tests PRIVATE
|
|
|
|
|
_c_dataengine pybind11::embed
|
|
|
|
|
${GTEST_LIBRARY}
|
|
|
|
|
${SECUREC_LIBRARY}
|
|
|
|
|
${SLOG_LIBRARY}
|
|
|
|
|
)
|
2020-03-27 14:49:12 +08:00
|
|
|
|
|
|
|
|
gtest_discover_tests(de_ut_tests WORKING_DIRECTORY ${Project_DIR}/tests/dataset)
|
|
|
|
|
|
|
|
|
|
install(TARGETS de_ut_tests
|
2020-06-19 02:33:08 +08:00
|
|
|
RUNTIME DESTINATION test)
|
2020-03-27 14:49:12 +08:00
|
|
|
|
|
|
|
|
# For internal testing only.
|
|
|
|
|
install(DIRECTORY ${Project_DIR}/tests/dataset/data/
|
2020-06-19 02:33:08 +08:00
|
|
|
DESTINATION test/data)
|
