PYTEST：UNITSEST错误对象没有属性＆＃x27; assert_called_once_with

Question

我正在用pytest编写单元测试，并在assert_called_once_with上获得错误。 我厌倦了使用Pytest文档中所示的相同方式，但似乎我缺少一些东西。

# Class which I am trying to mock. (./src/Trading.py)
class BaseTrade:
    def __init__(self, name):
        self.name = name

class Trade(BaseTrade):
    def __init__ (self, name):
        BaseTrade.__init__(self, name)

    def get_balance(self, value):
        # do calculation and return some value
        # for demo purpose hard-coding it
        return  value * 10


#unit test (./unitest/test_test.py
import mock
import unittest
import sys

sys.path.append("../src")

import Trading

class TestTradeClass(unittest.TestCase):
    @classmethod
    def setUpClass(self):
        self.expected_balance = 100

    @classmethod
    def tearDownClass(self):
        pass
    def test_trade(self):
        with mock.patch.object(Trading.Trade, 'get_balance', new = lambda self, x: (x * 10) ) as mock_method:
            obj = Trading.Trade("AAPL")
            value = obj.get_balance(10)
            assert value == 100
        mock_method.assert_called_once_with(100)

Error on mock_method.assert_called_once_with(100)
  AttributeError: 'function' object has no attribute 'assert_called_once_with'

Answer 1

我现在相信您想要 side_effect 。这怎么样？一个文件，假设 test.py ：

#!/usr/bin/env python
import unittest

import mock


class BaseTrade:
    def __init__(self, name):
        self.name = name


class Trade(BaseTrade):
    def __init__(self, name):
        BaseTrade.__init__(self, name)

    def get_balance(self, value):
        # do calculation and return some value
        # for demo purpose hard-coding it
        return value * 10


class TestTradeClass(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.expected_balance = 100

    def test_trade(self):
        # Without mock
        obj = Trade("AAPL")
        value = obj.get_balance(10)
        assert value == 100

        # With Trade.get_balance param patched
        with mock.patch.object(
            Trade, 'get_balance', side_effect=lambda value: value * 11
        ) as mock_method:
            obj = Trade("AAPL")
            value = obj.get_balance(10)
            assert value == 110
        mock_method.assert_called_once_with(10)


if __name__ == "__main__":
    unittest.main()

chmod +x test.py

./ test.py

输出：

.
----------------------------------------------------------------------
Ran 1 test in 0.000s

OK

说明：

• 使用 side> side_fect < /code>而不是新
• 组合到一个文件中，以使其更容易
• 删除 trade.trade
• @classmethod 使用 cls 而不是 self 。
• MOCK_METHOD.ASSERT_CALLED_ONCE_WITH（10），AS side_feft 关心通过 obj.get_balance（10）和存在以更改输出的值。

更近？如果没有，您可以澄清您要嘲笑什么？

Answer 2

这并不容易，但是如果 trade.trade.get_method（）实际上是一种香草功能，而不是一种方法 - 您可能需要 unittest.mock.create_autospec（）

是否有任何帮助？

• 'function'对象没有属性'assert_called_once_with'
• python3模拟替换功能使用另一个函数

不清楚whaat trade.trade.trade.trade 是。

如果偶然，交易是一个具有 self.trade = Trade（...）内部的类，您的问题将大不相同。然后，您需要进行更深入的修补 trade.get_method 。您可能想从同一模块中导入交易类交易使用它（例如，从.trading Import Trading，Trade Trade ） - 不是从其中 trade 从 - 然后修补 trade.get_method 。