Halaman utama Jelajahi Bantuan
Daftar Masuk
dayuan
/
manyi
6
0
Fork 0
Berkas Masalah 0 Permintaan Tarik 0 Wiki
Pohon: 34216f9fac
Ranting Tag
manyi
/
venv
/
Lib
/
site-packages
/
scipy
/
interpolate
/
tests
/
test_pade.py

test_pade.py 2.2 KB
Riwayat Mentahan

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566 
  1. from __future__ import division, print_function, absolute_import
    2. 

  2. 

  3. from numpy.testing import (assert_array_equal, assert_array_almost_equal)
    4. 

  4. from scipy.interpolate import pade
    5. 

  5. 

  6. def test_pade_trivial():
    7. 

  7.     nump, denomp = pade([1.0], 0)
    8. 

  8.     assert_array_equal(nump.c, [1.0])
    9. 

  9.     assert_array_equal(denomp.c, [1.0])
    10. 

  10.     
    11. 

  11.     nump, denomp = pade([1.0], 0, 0)
    12. 

  12.     assert_array_equal(nump.c, [1.0])
    13. 

  13.     assert_array_equal(denomp.c, [1.0])
    14. 

  14. 

  15. 

  16. def test_pade_4term_exp():
    17. 

  17.     # First four Taylor coefficients of exp(x).
    18. 

  18.     # Unlike poly1d, the first array element is the zero-order term.
    19. 

  19.     an = [1.0, 1.0, 0.5, 1.0/6]
    20. 

  20. 

  21.     nump, denomp = pade(an, 0)
    22. 

  22.     assert_array_almost_equal(nump.c, [1.0/6, 0.5, 1.0, 1.0])
    23. 

  23.     assert_array_almost_equal(denomp.c, [1.0])
    24. 

  24. 

  25.     nump, denomp = pade(an, 1)
    26. 

  26.     assert_array_almost_equal(nump.c, [1.0/6, 2.0/3, 1.0])
    27. 

  27.     assert_array_almost_equal(denomp.c, [-1.0/3, 1.0])
    28. 

  28. 

  29.     nump, denomp = pade(an, 2)
    30. 

  30.     assert_array_almost_equal(nump.c, [1.0/3, 1.0])
    31. 

  31.     assert_array_almost_equal(denomp.c, [1.0/6, -2.0/3, 1.0])
    32. 

  32. 

  33.     nump, denomp = pade(an, 3)
    34. 

  34.     assert_array_almost_equal(nump.c, [1.0])
    35. 

  35.     assert_array_almost_equal(denomp.c, [-1.0/6, 0.5, -1.0, 1.0])
    36. 

  36.    
    37. 

  37.     # Testing inclusion of optional parameter
    38. 

  38.     nump, denomp = pade(an, 0, 3)
    39. 

  39.     assert_array_almost_equal(nump.c, [1.0/6, 0.5, 1.0, 1.0])
    40. 

  40.     assert_array_almost_equal(denomp.c, [1.0])
    41. 

  41. 

  42.     nump, denomp = pade(an, 1, 2)
    43. 

  43.     assert_array_almost_equal(nump.c, [1.0/6, 2.0/3, 1.0])
    44. 

  44.     assert_array_almost_equal(denomp.c, [-1.0/3, 1.0])
    45. 

  45. 

  46.     nump, denomp = pade(an, 2, 1)
    47. 

  47.     assert_array_almost_equal(nump.c, [1.0/3, 1.0])
    48. 

  48.     assert_array_almost_equal(denomp.c, [1.0/6, -2.0/3, 1.0])
    49. 

  49. 

  50.     nump, denomp = pade(an, 3, 0)
    51. 

  51.     assert_array_almost_equal(nump.c, [1.0])
    52. 

  52.     assert_array_almost_equal(denomp.c, [-1.0/6, 0.5, -1.0, 1.0])
    53. 

  53.    
    54. 

  54.     # Testing reducing array
    55. 

  55.     nump, denomp = pade(an, 0, 2)
    56. 

  56.     assert_array_almost_equal(nump.c, [0.5, 1.0, 1.0])
    57. 

  57.     assert_array_almost_equal(denomp.c, [1.0])
    58. 

  58. 

  59.     nump, denomp = pade(an, 1, 1)
    60. 

  60.     assert_array_almost_equal(nump.c, [1.0/2, 1.0])
    61. 

  61.     assert_array_almost_equal(denomp.c, [-1.0/2, 1.0])
    62. 

  62. 

  63.     nump, denomp = pade(an, 2, 0)
    64. 

  64.     assert_array_almost_equal(nump.c, [1.0])
    65. 

  65.     assert_array_almost_equal(denomp.c, [1.0/2, -1.0, 1.0])
    66. 

  66.