Download test input to tests/data/

Author: mbrickerd

aoc/models/file.py

@@ -213,10 +213,10 @@ def add_test_input(day: int, part_num: int) -> None:
             part_num (int): The puzzle part number (1 or 2).
 
         Note:
-            Creates file at: `data/dayXX/test_XX_input.txt`
+            Creates file at: `tests/data/dayXX/test_XX_input.txt`
         """
         path = File.get_path()
-        folder = os.path.realpath(f"{path}/data/day{day:02}")
+        folder = os.path.realpath(f"{path}/tests/data/day{day:02}")
         folder_path = Path(folder)
         if not folder_path.exists():
             folder_path.mkdir(parents=True, exist_ok=True)

aoc/models/reader.py

@@ -62,9 +62,9 @@ def get_test_input(day: int, is_raw: bool, part_num: int) -> List[str]:
             List[str]: List of test input lines, processed according to `is_raw` flag.
 
         Note:
-            Expects input file at: `data/dayXX/test_YY_input.txt`
+            Expects input file at: `tests/data/dayXX/test_YY_input.txt`
             where `XX` is the zero-padded day number and `YY` is the zero-padded part number.
         """
-        file_path = os.path.join(Reader.PROJECT_ROOT, f"data/day{day:02d}/test_{part_num:02d}_input.txt")
+        file_path = os.path.join(Reader.PROJECT_ROOT, f"tests/data/day{day:02d}/test_{part_num:02d}_input.txt")
         with open(file_path, "r") as file:
             return [line.strip("\n") if is_raw else line.strip() for line in file.readlines()]