get a list of all cases, directories in validation datasets
(calculation_type: str, test_cases: list[str] | None = None)
| 125 | |
| 126 | |
| 127 | def get_test_case_paths(calculation_type: str, test_cases: list[str] | None = None) -> dict[str, Path]: |
| 128 | """get a list of all cases, directories in validation datasets""" |
| 129 | calculation_type_dir = DATA_PATH / calculation_type |
| 130 | test_case_paths = { |
| 131 | str(item.relative_to(DATA_PATH)).replace("\\", "/"): item |
| 132 | for item in calculation_type_dir.glob("**/") |
| 133 | if (item.is_dir() and (item / "params.json").is_file()) |
| 134 | } |
| 135 | if test_cases is not None: |
| 136 | test_case_paths = {key: value for key, value in test_case_paths.items() if key in test_cases} |
| 137 | |
| 138 | return test_case_paths |
| 139 | |
| 140 | |
| 141 | def add_case( |