def test_modifyCfgs_modifesFlair_combination(self): print("ARRANGE - MODIFYCFGS (MODIFIESFLAIR_COMBINATION)") print("-" * 20) # make test data folder cwd = setup.os.getcwd() print("Current directory:", cwd) setup.os.chdir(PACKAGE_PATH) print("Changed directory to:", PACKAGE_PATH) print("Creating data directory") subjects = ["sub-01", "sub-02", "sub-03"] for subject in subjects: setup.os.mkdir("data/" + subject) setup.os.mkdir("data/" + subject + "/anat") setup.os.system("touch data/sub-01/anat/sub-01_FLAIR.nii.gz") setup.os.system("touch data/sub-01/anat/sub-01_T1W.nii.gz") setup.os.system("touch data/sub-02/anat/sub-02_FLAIR.nii.gz") setup.os.system("touch data/sub-03/anat/sub-03_T1W.nii.gz") setup.os.chdir(cwd) print("Changed directory to:", cwd) print("ACT - MODIFYCFGS (MODIFIESFLAIR_COMBINATION)") print("-" * 20) inference.modifyCfgs(DATA_PATH) print("ASSERT - MODIFYCFGS (MODIFIESFLAIR_COMBINATION)") print("-" * 20) path_flair = str( PACKAGE_PATH ) + "/deepmedic/inference_model/config/test/test_flair.cfg" with open(path_flair) as f: content = f.read() output = content.split("\n") output.pop() expected = [DATA_PATH + "/sub-01/anat/sub-01_FLAIR.nii.gz"] self.assertEqual(expected, output)
def test_modifyCfgs_modifesT1w_all(self): print("ARRANGE - MODIFYCFGS (MODIFIEST1W_ALL)") print("-" * 20) # make test data folder cwd = setup.os.getcwd() print("Current directory:", cwd) setup.os.chdir(PACKAGE_PATH) print("Changed directory to:", PACKAGE_PATH) print("Creating data directory") subjects = ["sub-01", "sub-02", "sub-03"] for subject in subjects: setup.os.mkdir("data/" + subject) setup.os.mkdir("data/" + subject + "/anat") setup.os.system("touch data/" + subject + "/anat/" + subject + "_FLAIR.nii.gz") setup.os.system("touch data/" + subject + "/anat/" + subject + "_T1W.nii.gz") setup.os.chdir(cwd) print("Changed directory to:", cwd) print("ACT - MODIFYCFGS (MODIFIEST1W_ALL)") print("-" * 20) inference.modifyCfgs(DATA_PATH) print("ASSERT - MODIFYCFGS (MODIFIEST1W_ALL)") print("-" * 20) path_t1w = str( PACKAGE_PATH ) + "/deepmedic/inference_model/config/test/test_t1w.cfg" with open(path_t1w) as f: content = f.read() output = content.split("\n") output.pop() expected = [] for subject in subjects: expected.append(DATA_PATH + "/" + subject + "/anat/" + subject + "_T1W.nii.gz") self.assertEqual(expected, output)