在這個 class 中測試這個“setUp()”function 的正確方法是什么？

Question

它來自《L Earn Algorithmic Trading》一書——Sebastien Donadio 和 Sourav Ghosh

src 完整代碼>> https://github.com/PacktPublishing/Learn-Algorithmic-Trading/tree/master/Chapter7

我應該測試

交易策略.py

有了這個

TradingStrategy_ut.py

從TradingStrategy_ut.py我必須調用setUp() function 來初始化TradingStrategy class 。

class TestMarketSimulator(unittest.TestCase):

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

我已經嘗試過，但我不明白如何使它工作，因為TradingStrategy.py文件中的TradingStrategy class需要3 個參數。

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

每次我調用 setUp() function 我都會遇到這個錯誤：

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

解決方案：

python3 -m unittest TradingStrategy_ut.py

還有：編輯TradingStrategy.py

在每個參數后添加None

  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