def test_exp_vector(self):
var = Variable(5, np.array([1.5, 5.9]))
x = Variable(3, np.array([5, 6]))
y = Variable(1, np.array([-4, 3]))
result = var.exp()
equation = x + y**2
equation_result = equation.exp()
self.assertAlmostEqual(np.exp(5), result.value)
self.assertAlmostEqual(np.exp(equation.value), equation_result.value)
np.testing.assert_array_almost_equal(
np.exp(5) * np.array([1.5, 5.9]), result.derivative)
np.testing.assert_array_almost_equal(
np.exp(equation.value) * equation.derivative,
equation_result.derivative)
def test_exp_scalar(self):
var = Variable(5, 1.5)
result = var.exp()
self.assertAlmostEqual(np.exp(5), result.value)
self.assertAlmostEqual(np.exp(5) * 1.5, result.derivative)