From 1f294c7f13f64fad8617f923a734498f4f3e0941 Mon Sep 17 00:00:00 2001 From: SWHL Date: Wed, 1 Nov 2023 08:18:01 +0800 Subject: [PATCH] Update unit testing --- tests/test_coco_to_labelImg.py | 2 +- tests/test_labelImg_to_publaynet.py | 16 ++++++++-------- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/tests/test_coco_to_labelImg.py b/tests/test_coco_to_labelImg.py index e665363..92f522d 100644 --- a/tests/test_coco_to_labelImg.py +++ b/tests/test_coco_to_labelImg.py @@ -20,7 +20,7 @@ def test_normal(): converter = COCO2labelImg(data_dir) converter() - save_dir: Path = test_file_dir / "COCO_labelImg_format" + save_dir: Path = test_file_dir / "YOLOV5_COCO_format_labelImg_format" assert save_dir.exists() train_dir: Path = save_dir / "train" diff --git a/tests/test_labelImg_to_publaynet.py b/tests/test_labelImg_to_publaynet.py index a163843..ac51dbc 100644 --- a/tests/test_labelImg_to_publaynet.py +++ b/tests/test_labelImg_to_publaynet.py @@ -18,22 +18,22 @@ def test_normal(): - data_dir = test_file_dir / 'labelImg_dataset' + data_dir = test_file_dir / "labelImg_dataset" convert(data_dir) - new_data_dir = test_file_dir / 'labelImg_dataset_publaynet' + new_data_dir = test_file_dir / "labelImg_dataset_publaynet" - train_data_dir = new_data_dir / 'train' + train_data_dir = new_data_dir / "train" assert train_data_dir.exists() - train_json_path = new_data_dir / 'train.json' + train_json_path = new_data_dir / "train.json" assert train_json_path.exists() - train_data_dir = new_data_dir / 'test' - assert train_data_dir.exists() + test_data_dir = new_data_dir / "test" + assert test_data_dir.exists() - train_json_path = new_data_dir / 'test.json' - assert train_json_path.exists() + test_json_path = new_data_dir / "test.json" + assert test_json_path.exists() shutil.rmtree(new_data_dir)