Prepare the groundwork for testing cases (#336)

This is to prevent CI from failing when modifying the benchmarks to test
#334

Author: olejandro

utils/run_benchmarks.py

@@ -215,6 +215,8 @@ def run_benchmark(
         args += ["--include_dummy_imports"]
     if "regions" in benchmark:
         args.extend(["--regions", benchmark["regions"]])
+    if "case" in benchmark:
+        args.extend(["--case", benchmark["case"]])
     if verbose:
         args.append("--verbose")
     if "inputs" in benchmark:

xl2times/datatypes.py

@@ -263,6 +263,7 @@ class TimesModel:
     custom_psets: DataFrame = field(default_factory=DataFrame)
     user_psets: DataFrame = field(default_factory=DataFrame)
     user_csets: DataFrame = field(default_factory=DataFrame)
+    cases: dict[str, str] = field(default_factory=dict)
 
     @property
     def external_regions(self) -> set[str]:
@@ -361,6 +362,8 @@ class Config:
     ie_override_in_syssettings: bool = False
     # Switch to include dummy imports in the model
     include_dummy_imports: bool
+    # Name of the case to produce dd files for
+    produce_case: str | None
 
     def __init__(
         self,
@@ -371,6 +374,7 @@ def __init__(
         veda_attr_defaults_file: str,
         regions: str,
         include_dummy_imports: bool,
+        case: str | None,
     ):
         self.times_xl_maps = Config._read_mappings(mapping_file)
         (
@@ -403,6 +407,7 @@ def __init__(
         self.times_xl_maps = list(name_to_map.values())
         self.filter_regions = Config._read_regions_filter(regions)
         self.include_dummy_imports = include_dummy_imports
+        self.produce_case = case
 
     @staticmethod
     def _read_times_sets(

xl2times/main.py

@@ -75,6 +75,7 @@ def read_xl(
     inputs: list[str],
     regions: str,
     include_dummy_imports: bool,
+    case: str | None = None,
     output_dir: str | None = None,
     no_cache: bool = False,
     stop_after_read: bool = False,
@@ -90,6 +91,7 @@ def read_xl(
         "veda-attr-defaults.json",
         regions,
         include_dummy_imports,
+        case,
     )
 
     if len(inputs) == 1:
@@ -573,6 +575,7 @@ def run(args: argparse.Namespace) -> str | None:
             args.input,
             args.regions,
             args.include_dummy_imports,
+            case=args.case,
             output_dir=args.output_dir,
             no_cache=args.no_cache,
             stop_after_read=True,
@@ -583,6 +586,7 @@ def run(args: argparse.Namespace) -> str | None:
         args.input,
         args.regions,
         args.include_dummy_imports,
+        case=args.case,
         output_dir=args.output_dir,
         no_cache=args.no_cache,
     )
@@ -633,6 +637,11 @@ def parse_args(arg_list: None | list[str]) -> argparse.Namespace:
         action="store_true",
         help="Include dummy import processes in the model",
     )
+    args_parser.add_argument(
+        "--case",
+        type=str,
+        help="Case name to produce dd files for. If not provided, all the input files are used",
+    )
     args_parser.add_argument(
         "--output_dir", type=str, default="output", help="Output directory"
     )