def test_duplicate_axis(self):
with self.assertRaises(ValueError):
functions.Prod((0, 0))
def test_invalid_axis_type(self):
with self.assertRaises(TypeError):
functions.Prod([0])
def test_invalid_axis_type_in_tuple(self):
with self.assertRaises(TypeError):
functions.Prod((1, 'x'))
def check_backward(self, x_data, y_grad):
gradient_check.check_backward(functions.Prod(self.axis, self.keepdims),
x_data,
y_grad,
atol=1e-3,
dtype=numpy.float64)
