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 )