skip_test parameter on initialisation

Author: mbrickerd

aoc/models/tester.py

@@ -36,7 +36,7 @@ def run_test(
 
         Note:
             - Uses the `initialise()` function to dynamically load the solution class
-            - Loads test input from `data/dayXX/test_YY_input.txt`
+            - Loads test input from `tests/data/dayXX/test_YY_input.txt`
             - Expects solution classes to have `part1()` and `part2()` methods
 
         Example:
@@ -45,7 +45,7 @@ def run_test(
             # Raises AssertionError if result != 42
         """
         # Instantiate the solution object with the necessary parameters
-        solution = initialise(day)
+        solution = initialise(day, skip_test=False)
 
         # Dynamically select the part method based on `part_num`
         part_method = getattr(solution, f"part{part_num}")

aoc/utils/initalise.py

@@ -1,35 +1,31 @@
 from importlib import import_module
 
 
-def initialise(day: int):
+def initialise(day: int, skip_test: bool = True):
     """
     Dynamically load and instantiate the `Solution` class for a specific puzzle day.
 
-    Creates a `Solution` instance for the specified day by dynamically importing
-    the appropriate module from the solutions package.
-
     Args:
         day (int): The day number (1-25) of the puzzle to initialize.
+        skip_test (bool, optional): Whether to skip test input and use puzzle input.
+                                  Defaults to True.
 
     Returns:
-        Solution: An instance of the day's `Solution` class, initialized with the day number.
+        Solution: An instance of the day's `Solution` class, initialized with the
+                 specified parameters.
 
     Note:
         - Expects solution modules to be named `dayXX.py` where `XX` is zero-padded day number
         - Expects each solution module to have a `Solution` class
         - Solution modules should be in the `solutions` package
-        - Solution class must accept `day` parameter in constructor
+        - Solution class must accept `day` and `skip_test` parameters in constructor
 
     Example:
-        >>> solution = initialise(1)
-        # Imports `solutions.day01` and returns `Solution(day=1)`
+        >>> solution = initialise(1)  # Uses puzzle input
+        >>> test_solution = initialise(1, skip_test=False)  # Uses test input
     """
-    # Dynamically import the solution module based on the day
     module_name = f"solutions.day{day:02d}"
     solution_module = import_module(module_name)
-
-    # Get the `Solution` class
     Solution = getattr(solution_module, "Solution")
 
-    # Instantiate and return the `Solution` object
-    return Solution(day=day)
+    return Solution(day=day, skip_test=skip_test)