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)