Question

我试图测试一个pandas方法被一些值调用。

但是，仅仅通过应用@patch装饰器会导致修补方法在pandas中抛出ValueError，而实际方法则不然。我只是试图测试Stock.calc_sma正在调用基础pandas.rolling_mean函数。

我假设@patch装饰者基本上添加了一些＆＃34;魔法＆＃34;我修补的方法允许我检查函数是否被调用。如果是这种情况，为什么pandas.rolling_mean函数的行为是否相同，无论是修补还是未修补？

应用程序/ models.py

import pandas as pd
class Stock:  # i've excluded a bunch of class methods, including the one that sets self.data, which is a DataFrame of stock prices.
    def calc_sma(self, num_days)
        if self.data.shape[0] > num_days:  # Stock.data holds a DataFrame of stock prices
                column_title = 'sma' + str(num_days)
                self.data[column_title] = pd.rolling_mean(self.data['Adj Close'], num_days)

应用程序/测试/ TestStockModel.py

def setUp(self):
    self.stock = MagicMock(Stock)
    self.stock.ticker = "AAPL"
    self.stock.data = DataFrame(aapl_test_data.data)

@patch('app.models.pd.rolling_mean')
def test_calc_sma(self, patched_rolling_mean):
    Stock.calc_sma(self.stock, 3)
    assert(isinstance(self.stock.data['sma3'], Series))
    patched_rolling_mean.assert_any_call()

错误：test_calc_sma（TestStockModel.TestStockModel）

Traceback (most recent call last):
  File "/Users/grant/Code/python/chartflux/env/lib/python2.7/site-packages/mock.py", line 1201, in patched
    return func(*args, **keywargs)
  File "/Users/grant/Code/python/chartflux/app/tests/TestStockModel.py", line 26, in test_calc_sma
    Stock.calc_sma(self.stock, 3)
  File "/Users/grant/Code/python/chartflux/app/models.py", line 27, in calc_sma
    self.data[column_title] = pd.rolling_mean(self.data['Adj Close'], num_days)
  File "/Users/grant/Code/python/chartflux/env/lib/python2.7/site-packages/pandas/core/frame.py", line 1887, in __setitem__
    self._set_item(key, value)
  File "/Users/grant/Code/python/chartflux/env/lib/python2.7/site-packages/pandas/core/frame.py", line 1967, in _set_item
    value = self._sanitize_column(key, value)
  File "/Users/grant/Code/python/chartflux/env/lib/python2.7/site-packages/pandas/core/frame.py", line 2017, in _sanitize_column
    raise ValueError('Length of values does not match length of '
ValueError: Length of values does not match length of index

Answer 1

>>> import os
>>> os.getcwd()
'/'
>>> from unittest.mock import patch
>>> with patch('os.getcwd'):
...     print(os.getcwd)
...     print(os.getcwd())
...     print(len(os.getcwd()))
...
<MagicMock name='getcwd' id='4472112296'>
<MagicMock name='getcwd()' id='4472136928'>
0

默认情况下，patch用真正的通用模拟对象替换。如您所见，调用mock只会返回另一个模拟。即使被替换的对象没有len，它的len也为0。它的属性也是通用的模拟。

因此，模拟行为需要额外的参数，如：

>>> with patch('os.getcwd', return_value='/a/wonderful/place'):
...     os.getcwd()
...
'/a/wonderful/place'

或“通过”：

>>> _cwd = os.getcwd
>>> with patch('os.getcwd') as p:
...     p.side_effect = lambda: _cwd()
...     print(os.getcwd())
...
/

https://docs.python.org/3.5/library/unittest.mock-examples.html

中有类似的例子

修改方法而不改变方法的工作方式？

错误：test_calc_sma（TestStockModel.TestStockModel）

1 个答案: