(self, input_params)
| 221 | @parameterized.expand(SIM_GPU_TEST_CASES) |
| 222 | @skip_if_no_cuda |
| 223 | def test_data_analyzer_gpu(self, input_params): |
| 224 | sim_dim = input_params["sim_dim"] |
| 225 | label_key = input_params["label_key"] |
| 226 | image_only = not bool(label_key) |
| 227 | rmax = max(int(sim_dim[0] / 4), 1) |
| 228 | create_sim_data( |
| 229 | self.dataroot_dir, sim_datalist, sim_dim, image_only=image_only, rad_max=rmax, rad_min=1, num_seg_classes=1 |
| 230 | ) |
| 231 | analyser = DataAnalyzer( |
| 232 | self.datalist_file, self.dataroot_dir, output_path=self.datastat_file, label_key=label_key, device="cuda" |
| 233 | ) |
| 234 | datastat = analyser.get_all_case_stats() |
| 235 | |
| 236 | assert len(datastat["stats_by_cases"]) == len(sim_datalist["training"]) |
| 237 | |
| 238 | def test_basic_operation_class(self): |
| 239 | op = TestOperations() |
nothing calls this directly
no test coverage detected