Broker add deposit and withdraw functions

kernc · AdiSai · Mar 2, 2023 · Mar 2, 2023 · Mar 2, 2023 · a473d4c8cef06ac27b801aa8f016a8b90e12bab3
commit a473d4c8cef06ac27b801aa8f016a8b90e12bab3
diff --git a/backtesting/backtesting.py b/backtesting/backtesting.py
@@ -721,6 +721,16 @@ def __init__(self, *, data, cash, commission, margin,
 
     def __repr__(self):
         return f'<Broker: {self._cash:.0f}{self.position.pl:+.1f} ({len(self.trades)} trades)>'
+
+    def deposit_cash(self, amount: float):
+        self._cash += amount
+        return self._cash
+
+    def withdraw_cash(self, amount: float):
+        if amount > self._cash:
+            return self._cash
+        self.cash -= amount
+        return self._cash
 
     def new_order(self,
                   size: float,