(self)
| 106 | @unittest.skipIf(not has_tb, "no tensorboard summary writer") |
| 107 | class TestEnsembleBuilder(unittest.TestCase): |
| 108 | def setUp(self) -> None: |
| 109 | set_determinism(0) |
| 110 | self.test_dir = tempfile.TemporaryDirectory() |
| 111 | test_path = self.test_dir.name |
| 112 | |
| 113 | dataroot = os.path.join(test_path, "dataroot") |
| 114 | work_dir = os.path.join(test_path, "workdir") |
| 115 | |
| 116 | da_output_yaml = os.path.join(work_dir, "datastats.yaml") |
| 117 | data_src_cfg = os.path.join(work_dir, "data_src_cfg.yaml") |
| 118 | |
| 119 | if not os.path.isdir(work_dir): |
| 120 | os.makedirs(work_dir) |
| 121 | |
| 122 | create_sim_data(dataroot, fake_datalist, (24, 24, 24), rad_max=10, rad_min=1, num_seg_classes=1) |
| 123 | |
| 124 | # write to a json file |
| 125 | fake_json_datalist = os.path.join(dataroot, "fake_input.json") |
| 126 | ConfigParser.export_config_file(fake_datalist, fake_json_datalist) |
| 127 | |
| 128 | da = DataAnalyzer(fake_json_datalist, dataroot, output_path=da_output_yaml) |
| 129 | da.get_all_case_stats() |
| 130 | |
| 131 | data_src = { |
| 132 | "name": "fake_data", |
| 133 | "task": "segmentation", |
| 134 | "modality": "MRI", |
| 135 | "datalist": fake_json_datalist, |
| 136 | "dataroot": dataroot, |
| 137 | "multigpu": False, |
| 138 | "class_names": ["label_class"], |
| 139 | } |
| 140 | |
| 141 | ConfigParser.export_config_file(data_src, data_src_cfg) |
| 142 | |
| 143 | self.da_output_yaml = da_output_yaml |
| 144 | self.work_dir = work_dir |
| 145 | self.data_src_cfg_name = data_src_cfg |
| 146 | |
| 147 | def test_ensemble(self) -> None: |
| 148 | with skip_if_downloading_fails(): |
nothing calls this directly
no test coverage detected