31 lines
805 B
Python
31 lines
805 B
Python
import os
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
TEST_FOLDER = ".test_folder"
|
|
TEST_IMAGES = "tests/test_files"
|
|
TEST_TEMP_FOLDER = os.path.join(TEST_FOLDER, 'Temp')
|
|
TEST_IMAGE_FOLDER = os.path.join(TEST_FOLDER, 'Images')
|
|
|
|
|
|
def create_file(file):
|
|
Path(os.path.join(TEST_FOLDER, file)).touch()
|
|
|
|
|
|
def delete_folder():
|
|
shutil.rmtree(TEST_FOLDER, ignore_errors=True)
|
|
|
|
|
|
def create_folders():
|
|
delete_folder()
|
|
os.makedirs(os.path.join(TEST_FOLDER, '001', '001'))
|
|
os.makedirs(os.path.join(TEST_FOLDER, '002', '001'))
|
|
|
|
|
|
def copy_test_images():
|
|
delete_folder()
|
|
os.makedirs(TEST_TEMP_FOLDER)
|
|
os.makedirs(TEST_IMAGE_FOLDER)
|
|
shutil.copy(src=os.path.join(TEST_IMAGES, 'test_image_001.JPG'), dst=TEST_TEMP_FOLDER)
|
|
shutil.copy(src=os.path.join(TEST_IMAGES, 'test_image_002.JPG'), dst=TEST_TEMP_FOLDER)
|