def test_polyderiv(self): a = self.coefficients b = [ (i+1)*a[i+1] for i in range(len(a)-1) ] self.assertEqual(b, polyuv.polyderiv(a)) c = [ (i+1)*b[i+1] for i in range(len(b)-1) ] self.assertEqual(c, polyuv.polyderiv(b)) self.assertEqual(c, polyuv.polyderiv(a,d=2))
def test_deriv(self): coefs = range(6) random.shuffle(coefs) p = numpy.poly1d(list(reversed(coefs))) for n in range(len(coefs)): self.assertEqual(polyuv.polyderiv(coefs, n), list(p.deriv(m=n).c)[::-1] )