def test_invalid_axis(self): with self.assertRaises(TypeError): functions.Rollaxis('a', 0)
def test_invalid_start(self): with self.assertRaises(TypeError): functions.Rollaxis(0, 'a')
def check_backward(self, x_data, g_data): gradient_check.check_backward( functions.Rollaxis(self.axis, self.start), x_data, g_data)