Example #1
0
 def test_deriv(self):
     """Test ad.d() function"""
     z = self.y * self.x
     dz = ad.d(z, self.x)
     assert_allclose(dz, self.y)
     
     z = self.x_row ** 2
     dz = ad.d(z, self.x_row)
     assert_allclose(dz, [0.0, 1.0, 2.0, 3.0, 4.0])
     
     z = self.y * self.x_row
     dz = ad.d(z, self.x_row)
     assert_allclose(dz, self.y)
Example #2
0
 def test_d2(self):
     """Test ad.d2() function"""
     z = self.y * exp(-2*self.x)
     dz = ad.d(z, self.x)
     ddz = ad.d2(z, self.x)
     assert_allclose(dz, -2*z)
     assert_allclose(ddz, 4*z)
     
     z = self.x_row ** 2
     ddz = ad.d2(z, self.x_row)
     assert_allclose(ddz, 2.)
     
     z = self.y * sin(2*self.x_row)
     ddz = ad.d2(z, self.x_row)
     assert_allclose(ddz, -4*z)