test_precompute_expn_asy.py 649 B

1234567891011121314151617181920212223242526
  1. from __future__ import division, print_function, absolute_import
  2. from numpy.testing import assert_equal
  3. from scipy.special._testutils import check_version, MissingModule
  4. from scipy.special._precompute.expn_asy import generate_A
  5. try:
  6. import sympy
  7. from sympy import Poly
  8. except ImportError:
  9. sympy = MissingModule("sympy")
  10. @check_version(sympy, "1.0")
  11. def test_generate_A():
  12. # Data from DLMF 8.20.5
  13. x = sympy.symbols('x')
  14. Astd = [Poly(1, x),
  15. Poly(1, x),
  16. Poly(1 - 2*x),
  17. Poly(1 - 8*x + 6*x**2)]
  18. Ares = generate_A(len(Astd))
  19. for p, q in zip(Astd, Ares):
  20. assert_equal(p, q)