diff --git a/src/fleet.py b/src/fleet.py index de639f3..d8926d4 100644 --- a/src/fleet.py +++ b/src/fleet.py @@ -6,8 +6,8 @@ from position import Position from custom_exception.nothing_to_compare_error import NothingToCompareError class Fleet: - def __init__(self, year: int, loader: BoatsLoader): - self.boats = loader.load_boats(year) + def __init__(self, year: int, loader = None, boats = None): + self.boats = loader.load_boats(year) if boats is None else boats def compare_boats(self, getter: callable, comparator: callable):