def mock_global_config(mocker: MockerFixture):
STORAGE_FOLDER = pytest.data_dir / 'meta-extraction' / 'storage_folder'
PICKLE_FOLDER = pytest.out_dir / 'meta-extraction' / 'pickles'
pytest.create_dirs(STORAGE_FOLDER, PICKLE_FOLDER)
return mocker.patch.multiple(
MetadataExtractor,
STORAGE_FOLDER=str(STORAGE_FOLDER),
PICKLE_FOLDER=str(PICKLE_FOLDER) + "/"
)
def create_out_dir_structure(out_dir: PurePath):
"""
Creates directory structure for cold-extraction output
"""
pytest.create_dirs(*[
out_dir / 'extracted-images', out_dir / 'failed-dicom/1', out_dir /
'failed-dicom/2', out_dir / 'failed-dicom/3', out_dir /
'failed-dicom/4', out_dir / 'maps', out_dir / 'meta'
])
return out_dir
def test_stop_success(self):
"""
Tests Success for dcm_anonymize function with params
dcm_folder, dcm_output_dir and stop flag.
"""
dcm_root_dir = test_config.input_dir / 'dcm_root_dir'
dcm_output_dir = test_config.output_dir / 'dcm_root_dir_out'
pytest.create_dirs(dcm_output_dir)
dcm_folders = DCMAnon.get_dcm_folders(dcm_root_dir)
try:
DCMAnon.dcm_anonymize(dcm_folders, dcm_output_dir, stop=2)
except SystemExit:
pass
except Exception as e:
pytest.fail(e, pytrace=True)
def setup_method(self):
header_list = [ImageExtractor.extract_headers((0, self.test_dcm_file))]
self.file_data = pd.DataFrame(header_list)
self.index = 0
self.invalid_file_data = pd.DataFrame([{
'some_col_1':
'Dummy Col1 Value',
'some_col_2':
'Dummy Col2 Value',
'file':
self.invalid_test_dcm_file
}])
out_dir = pytest.out_dir / 'png-extraction/outputs/TestExtractImages'
self.png_destination = f"{str(out_dir)}/extracted-images/"
self.failed = f"{str(out_dir)}/failed-dicom/"
pytest.create_dirs(out_dir, self.png_destination, self.failed)
def __init__(self):
pytest.create_dirs(
self.storage_folder,
self.input_dir
)
def __init__(self):
pytest.create_dirs(
self.input_dir
)
