diff --git a/tests/links_tests/model_tests/fpn_tests/test_faster_rcnn.py b/tests/links_tests/model_tests/fpn_tests/test_faster_rcnn.py index ac831d0020..d10c2a8a9b 100644 --- a/tests/links_tests/model_tests/fpn_tests/test_faster_rcnn.py +++ b/tests/links_tests/model_tests/fpn_tests/test_faster_rcnn.py @@ -1,9 +1,11 @@ from __future__ import division +import mock import numpy as np import unittest import chainer +from chainer import backends from chainer import testing from chainer.testing import attr @@ -46,6 +48,16 @@ def __init__(self, n_fg_class, return_values, min_size, max_size): ) +def dummy_roi_average_align_2d( + x, rois, roi_indices, outsize, spatial_scale, sampling_ratio=None): + if not isinstance(outsize, chainer.utils.collections_abc.Iterable): + outsize = outsize, outsize + + xp = backends.cuda.get_array_module(x.array) + y = _random_array(xp, (len(rois), x.shape[1], outsize[0], outsize[1])) + return chainer.Variable(y) + + @testing.parameterize(*testing.product_dict( [ {'return_values': 'detection'}, @@ -122,7 +134,9 @@ def _check_call(self): @attr.slow def test_call_cpu(self): - self._check_call() + with mock.patch('chainer.functions.roi_average_align_2d', + dummy_roi_average_align_2d): + self._check_call() @attr.gpu def test_call_gpu(self): @@ -155,7 +169,9 @@ def _check_predict(self): @attr.slow def test_predict_cpu(self): - self._check_predict() + with mock.patch('chainer.functions.roi_average_align_2d', + dummy_roi_average_align_2d): + self._check_predict() @attr.gpu def test_predict_gpu(self):