Merge pull request #907 from Hakuyume/use-dummy-roi-align-in-unittest

Use dummy roi_average_align_2d in cpu tests

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):