public/aoc2021
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

type annotations

Browse Source
This commit is contained in:
Stefan Harmuth 2021-11-21 13:10:10 +01:00
parent 2b8a69c24b
commit af48c1a8fb
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
aoclib/day.py
View File

@ -1,5 +1,5 @@
import os
from typing import List, Any
from typing import List, Any, Type
from .tools import splitLine
from .output import printSolution
@ -13,7 +13,7 @@ class AOCDay:
test_solutions_p1: List
test_solutions_p2: List
def __init__(self, day):
def __init__(self, day: int):
self.day = day
with open(os.path.join(INPUTS_PATH, "input%02d" % day)) as f:
self.input = f.read().splitlines()
@ -52,13 +52,13 @@ class AOCDay:
self.input = live_input
return True
def getInputListAsType(self, return_type):
def getInputListAsType(self, return_type: Type) -> List:
"""
get input as list casted to return_type, each line representing one list entry
"""
return [return_type(i) for i in self.input]
def getMultiLineInputAsArray(self, return_type=None, join_char=None):
def getMultiLineInputAsArray(self, return_type: Type = None, join_char: str = None) -> List:
"""
get input for day x as 2d array, split by empty lines
"""
@ -83,7 +83,7 @@ class AOCDay:
return return_array
def getInputAs2DArray(self, return_type=None):
def getInputAs2DArray(self, return_type: Type = None) -> List:
"""
get input for day x as 2d-array (a[line][pos])
"""
@ -96,7 +96,7 @@ class AOCDay:
return return_array
def getInputAsArraySplit(self, split_char=',', return_type=None):
def getInputAsArraySplit(self, split_char: str = ',', return_type: Type = None) -> List:
"""
get input for day x with the lines split by split_char
if input has only one line, returns a 1d array with the values