Update gas_station.py

Author: cclauss

Diff for: greedy_methods/gas_station.py

@@ -33,25 +33,31 @@ class GasStation:
     cost: int
 
 
-def get_gas_stations(gas_quantities: list[int], costs: list[int]) -> list[GasStation]:
+def get_gas_stations(gas_quantities: list[int], costs: list[int]) -> tuple[GasStation]:
     """
-    This function returns a list of gas stations.
+    This function returns a tuple of gas stations.
 
     Args:
-        gas_quantities [list]: Amount of gas available at each station
-        cost [list]: The cost of gas required to move from a station to the next
+        gas_quantities: Amount of gas available at each station
+        costs: The cost of gas required to move from a station to the next
 
     Returns:
-        gas_stations [list]: a list of gas stations
+        A tuple of gas stations
+    
+    >>> gas_stations = get_gas_stations([1, 2, 3, 4, 5], [3, 4, 5, 1, 2])
+    >>> len(gas_stations)
+    5
+    >>> gas_stations[0]
+    GasStation(gas_quantity=1, cost=3)
+    >>> gas_stations[-1]
+    GasStation(gas_quantity=5, cost=2)
     """
-    gas_stations = [
-        GasStation(gas_quantity, cost)
-        for (gas_quantity, cost) in zip(gas_quantities, costs)
-    ]
-    return gas_stations
+    return tuple(
+        GasStation(quantity, cost) for quantity, cost in zip(gas_quantities, costs)
+    )
 
 
-def can_complete_journey(gas_quantities: list[int], costs: list[int]) -> int:
+def can_complete_journey(gas_stations: tuple[GasStation]) -> int:
     """
     This function returns the index from which to start the journey
     in order to reach the end.
@@ -64,22 +70,19 @@ def can_complete_journey(gas_quantities: list[int], costs: list[int]) -> int:
         start [int]: start index needed to complete the journey
 
     Examples:
-    >>> can_complete_journey([1, 2, 3, 4, 5], [3, 4, 5, 1, 2])
+    >>> can_complete_journey(get_gas_stations([1, 2, 3, 4, 5], [3, 4, 5, 1, 2]))
     3
-    >>> can_complete_journey([2, 3, 4], [3, 4, 3])
+    >>> can_complete_journey(get_gas_stations([2, 3, 4], [3, 4, 3]))
     -1
-
     """
-    total_gas = sum(gas_quantities)
-    total_cost = sum(costs)
+    total_gas = sum(gas_station.gas_quantity for gas_station in gas_stations)
+    total_cost = sum(gas_station.cost for gas_station in gas_stations)
 
     if total_gas < total_cost:
         return -1
 
     start = 0
     net = 0
-    gas_stations = get_gas_stations(gas_quantities, costs)
-
     for i, gas_station in enumerate(gas_stations):
         net += gas_station.gas_quantity - gas_station.cost
         if net < 0: