business class styling

pricing/dynamic/business.py

@@ -1,17 +1,19 @@
+from typing import List
 import pandas as pd
 from pandas import DataFrame
+from pricing.dynamic.customer import Customer
 
 
 class Business:
-    def __init__(self, costs, customer):
+    def __init__(self, costs: List[float], customer: Customer) -> None:
         self.costs = costs
         self.net_profit = 0
         self.prices = costs.copy()
         self.tiers = len(costs)
         self.customer = customer
         self.transaction_history = DataFrame(columns=['prices', 'chosen_tier'])
 
-    def sell(self):
+    def sell(self) -> None:
         tier = self.customer.choose_tier(self.costs, self.prices)
         self.net_profit += self.prices[tier] - self.costs[tier]
         new_transaction = pd.DataFrame([[self.prices, tier]],
@@ -20,6 +22,6 @@ def sell(self):
                                               self.transaction_history],
                                              ignore_index=True)
 
-    def sell_n(self, n):
+    def sell_n(self, n: int) -> None:
         for _ in range(n):
             self.sell()