在这个 class 中测试这个“setUp()”function 的正确方法是什么？

Question

It's from the book L earn Algorithmic Trading- Sebastien Donadio and Sourav Ghosh

src full code >> https://github.com/PacktPublishing/Learn-Algorithmic-Trading/tree/master/Chapter7

I'm supposed to test the

TradingStrategy.py

with this

TradingStrategy_ut.py

From the TradingStrategy_ut.py I have to call the setUp() function to initialize the TradingStrategy class .

class TestMarketSimulator(unittest.TestCase):

    def setUp(self):
        self.trading_strategy= TradingStrategy()

I've tried and I don't understand how to make it work because the TradingStrategy class that's in the TradingStrategy.py file takes 3 parameters.

class TradingStrategy:
    def __init__(self, ob_2_ts, ts_2_om, om_2_ts):
        self.orders = []
        self.order_id = 0
        self.position = 0
        self.pnl = 0
        self.cash = 10000
        self.current_bid = 0
        self.current_offer = 0
        self.ob_2_ts = ob_2_ts
        self.ts_2_om = ts_2_om
        self.om_2_ts = om_2_ts

and every time I call the setUp() function I have this error:

x = TestMarketSimulator()
x.setUp()

---

Traceback (most recent call last):
  File "TradingStrategy_ut.py", line 72, in <module>
    x.setUp()
  File "TradingStrategy_ut.py", line 9, in setUp
    self.trading_strategy= TradingStrategy()
TypeError: __init__() missing 3 required positional arguments: 'ob_2_ts', 'ts_2_om', and 'om_2_ts'

Answer 1

SOLUTION :

python3 -m unittest TradingStrategy_ut.py

and also: edited TradingStrategy.py

added None after every parameter

  class TradingStrategy:
        def __init__(self, ob_2_ts=None, ts_2_om=None, om_2_ts=None): 
            self.orders = []
            self.order_id = 0
            self.position = 0
            self.pnl = 0
            self.cash = 10000
            self.current_bid = 0
            self.current_offer = 0
            self.ob_2_ts = ob_2_ts
            self.ts_2_om = ts_2_om
            self.om_2_ts = om_2_ts