test_polyutils.py 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. """Tests for polyutils module.
  2. """
  3. import numpy as np
  4. import numpy.polynomial.polyutils as pu
  5. from numpy.testing import (
  6. assert_almost_equal, assert_raises, assert_equal, assert_,
  7. )
  8. class TestMisc:
  9. def test_trimseq(self):
  10. for i in range(5):
  11. tgt = [1]
  12. res = pu.trimseq([1] + [0]*5)
  13. assert_equal(res, tgt)
  14. def test_as_series(self):
  15. # check exceptions
  16. assert_raises(ValueError, pu.as_series, [[]])
  17. assert_raises(ValueError, pu.as_series, [[[1, 2]]])
  18. assert_raises(ValueError, pu.as_series, [[1], ['a']])
  19. # check common types
  20. types = ['i', 'd', 'O']
  21. for i in range(len(types)):
  22. for j in range(i):
  23. ci = np.ones(1, types[i])
  24. cj = np.ones(1, types[j])
  25. [resi, resj] = pu.as_series([ci, cj])
  26. assert_(resi.dtype.char == resj.dtype.char)
  27. assert_(resj.dtype.char == types[i])
  28. def test_trimcoef(self):
  29. coef = [2, -1, 1, 0]
  30. # Test exceptions
  31. assert_raises(ValueError, pu.trimcoef, coef, -1)
  32. # Test results
  33. assert_equal(pu.trimcoef(coef), coef[:-1])
  34. assert_equal(pu.trimcoef(coef, 1), coef[:-3])
  35. assert_equal(pu.trimcoef(coef, 2), [0])
  36. class TestDomain:
  37. def test_getdomain(self):
  38. # test for real values
  39. x = [1, 10, 3, -1]
  40. tgt = [-1, 10]
  41. res = pu.getdomain(x)
  42. assert_almost_equal(res, tgt)
  43. # test for complex values
  44. x = [1 + 1j, 1 - 1j, 0, 2]
  45. tgt = [-1j, 2 + 1j]
  46. res = pu.getdomain(x)
  47. assert_almost_equal(res, tgt)
  48. def test_mapdomain(self):
  49. # test for real values
  50. dom1 = [0, 4]
  51. dom2 = [1, 3]
  52. tgt = dom2
  53. res = pu.mapdomain(dom1, dom1, dom2)
  54. assert_almost_equal(res, tgt)
  55. # test for complex values
  56. dom1 = [0 - 1j, 2 + 1j]
  57. dom2 = [-2, 2]
  58. tgt = dom2
  59. x = dom1
  60. res = pu.mapdomain(x, dom1, dom2)
  61. assert_almost_equal(res, tgt)
  62. # test for multidimensional arrays
  63. dom1 = [0, 4]
  64. dom2 = [1, 3]
  65. tgt = np.array([dom2, dom2])
  66. x = np.array([dom1, dom1])
  67. res = pu.mapdomain(x, dom1, dom2)
  68. assert_almost_equal(res, tgt)
  69. # test that subtypes are preserved.
  70. class MyNDArray(np.ndarray):
  71. pass
  72. dom1 = [0, 4]
  73. dom2 = [1, 3]
  74. x = np.array([dom1, dom1]).view(MyNDArray)
  75. res = pu.mapdomain(x, dom1, dom2)
  76. assert_(isinstance(res, MyNDArray))
  77. def test_mapparms(self):
  78. # test for real values
  79. dom1 = [0, 4]
  80. dom2 = [1, 3]
  81. tgt = [1, .5]
  82. res = pu. mapparms(dom1, dom2)
  83. assert_almost_equal(res, tgt)
  84. # test for complex values
  85. dom1 = [0 - 1j, 2 + 1j]
  86. dom2 = [-2, 2]
  87. tgt = [-1 + 1j, 1 - 1j]
  88. res = pu.mapparms(dom1, dom2)
  89. assert_almost_equal(res, tgt)