def test_numpy_to_numpy_with_device(self):
orig = self.orig_numpy()
self.send_check_equal(orig, backend.CpuDevice())
def _generate_inputs(self):
inputs = self.generate_inputs()
_check_array_types(inputs, backend.CpuDevice(), 'generate_inputs')
return inputs
def test_eq_numpy(self):
assert backend.get_device(numpy) == backend.get_device(numpy)
assert backend.CpuDevice() == backend.get_device(numpy)
def send_check_equal(self, orig, device_spec):
device = backend.get_device(device_spec)
converted = device.send(orig)
numpy.testing.assert_array_equal(backend.CpuDevice().send(orig),
backend.CpuDevice().send(converted))
return converted
def check_unrecognized(self, arg):
device = backend.get_device_from_array(arg)
assert device == backend.CpuDevice()
def test_module_numpy_device(self):
self.check_device_spec_numpy(backend.CpuDevice())
def check_deserialize_by_passing_none(self, y, query):
ret = self.deserializer(query, None)
numpy.testing.assert_array_equal(
backend.CpuDevice().send(ret), self.data)
def test_init(self):
device = backend.CpuDevice()
self.check_device(device)
def check_deserialize(self, y, query):
ret = self.deserializer(query, y)
numpy.testing.assert_array_equal(
backend.CpuDevice().send(y), self.data)
self.assertIs(ret, y)
def check_deserialize_none_value(self, y):
ret = self.deserializer('y', None)
numpy.testing.assert_array_equal(backend.CpuDevice().send(ret),
self.data)
def setup_cpu(self):
self._setup(backend.CpuDevice())
self.f.forward_cpu = mock.MagicMock(return_value=(self.y1, self.y2))
def _forward_expected(self, inputs):
outputs = self.forward_expected(inputs)
_check_array_types(outputs, backend.CpuDevice(), 'forward_expected')
return outputs
def _generate_grad_grad_inputs(self, inputs_template):
grad_grad_inputs = self.generate_grad_grad_inputs(inputs_template)
_check_array_types(
grad_grad_inputs, backend.CpuDevice(), 'generate_grad_grad_inputs')
return grad_grad_inputs