Spaces:
Running
Running
File size: 4,520 Bytes
6a86ad5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 |
"""Tests for high-level polynomials manipulation functions. """
from sympy.polys.polyfuncs import (
symmetrize, horner, interpolate, rational_interpolate, viete,
)
from sympy.polys.polyerrors import (
MultivariatePolynomialError,
)
from sympy.core.singleton import S
from sympy.core.symbol import symbols
from sympy.testing.pytest import raises
from sympy.abc import a, b, c, d, e, x, y, z
def test_symmetrize():
assert symmetrize(0, x, y, z) == (0, 0)
assert symmetrize(1, x, y, z) == (1, 0)
s1 = x + y + z
s2 = x*y + x*z + y*z
assert symmetrize(1) == (1, 0)
assert symmetrize(1, formal=True) == (1, 0, [])
assert symmetrize(x) == (x, 0)
assert symmetrize(x + 1) == (x + 1, 0)
assert symmetrize(x, x, y) == (x + y, -y)
assert symmetrize(x + 1, x, y) == (x + y + 1, -y)
assert symmetrize(x, x, y, z) == (s1, -y - z)
assert symmetrize(x + 1, x, y, z) == (s1 + 1, -y - z)
assert symmetrize(x**2, x, y, z) == (s1**2 - 2*s2, -y**2 - z**2)
assert symmetrize(x**2 + y**2) == (-2*x*y + (x + y)**2, 0)
assert symmetrize(x**2 - y**2) == (-2*x*y + (x + y)**2, -2*y**2)
assert symmetrize(x**3 + y**2 + a*x**2 + b*y**3, x, y) == \
(-3*x*y*(x + y) - 2*a*x*y + a*(x + y)**2 + (x + y)**3,
y**2*(1 - a) + y**3*(b - 1))
U = [u0, u1, u2] = symbols('u:3')
assert symmetrize(x + 1, x, y, z, formal=True, symbols=U) == \
(u0 + 1, -y - z, [(u0, x + y + z), (u1, x*y + x*z + y*z), (u2, x*y*z)])
assert symmetrize([1, 2, 3]) == [(1, 0), (2, 0), (3, 0)]
assert symmetrize([1, 2, 3], formal=True) == ([(1, 0), (2, 0), (3, 0)], [])
assert symmetrize([x + y, x - y]) == [(x + y, 0), (x + y, -2*y)]
def test_horner():
assert horner(0) == 0
assert horner(1) == 1
assert horner(x) == x
assert horner(x + 1) == x + 1
assert horner(x**2 + 1) == x**2 + 1
assert horner(x**2 + x) == (x + 1)*x
assert horner(x**2 + x + 1) == (x + 1)*x + 1
assert horner(
9*x**4 + 8*x**3 + 7*x**2 + 6*x + 5) == (((9*x + 8)*x + 7)*x + 6)*x + 5
assert horner(
a*x**4 + b*x**3 + c*x**2 + d*x + e) == (((a*x + b)*x + c)*x + d)*x + e
assert horner(4*x**2*y**2 + 2*x**2*y + 2*x*y**2 + x*y, wrt=x) == ((
4*y + 2)*x*y + (2*y + 1)*y)*x
assert horner(4*x**2*y**2 + 2*x**2*y + 2*x*y**2 + x*y, wrt=y) == ((
4*x + 2)*y*x + (2*x + 1)*x)*y
def test_interpolate():
assert interpolate([1, 4, 9, 16], x) == x**2
assert interpolate([1, 4, 9, 25], x) == S(3)*x**3/2 - S(8)*x**2 + S(33)*x/2 - 9
assert interpolate([(1, 1), (2, 4), (3, 9)], x) == x**2
assert interpolate([(1, 2), (2, 5), (3, 10)], x) == 1 + x**2
assert interpolate({1: 2, 2: 5, 3: 10}, x) == 1 + x**2
assert interpolate({5: 2, 7: 5, 8: 10, 9: 13}, x) == \
-S(13)*x**3/24 + S(12)*x**2 - S(2003)*x/24 + 187
assert interpolate([(1, 3), (0, 6), (2, 5), (5, 7), (-2, 4)], x) == \
S(-61)*x**4/280 + S(247)*x**3/210 + S(139)*x**2/280 - S(1871)*x/420 + 6
assert interpolate((9, 4, 9), 3) == 9
assert interpolate((1, 9, 16), 1) is S.One
assert interpolate(((x, 1), (2, 3)), x) is S.One
assert interpolate({x: 1, 2: 3}, x) is S.One
assert interpolate(((2, x), (1, 3)), x) == x**2 - 4*x + 6
def test_rational_interpolate():
x, y = symbols('x,y')
xdata = [1, 2, 3, 4, 5, 6]
ydata1 = [120, 150, 200, 255, 312, 370]
ydata2 = [-210, -35, 105, 231, 350, 465]
assert rational_interpolate(list(zip(xdata, ydata1)), 2) == (
(60*x**2 + 60)/x )
assert rational_interpolate(list(zip(xdata, ydata1)), 3) == (
(60*x**2 + 60)/x )
assert rational_interpolate(list(zip(xdata, ydata2)), 2, X=y) == (
(105*y**2 - 525)/(y + 1) )
xdata = list(range(1,11))
ydata = [-1923885361858460, -5212158811973685, -9838050145867125,
-15662936261217245, -22469424125057910, -30073793365223685,
-38332297297028735, -47132954289530109, -56387719094026320,
-66026548943876885]
assert rational_interpolate(list(zip(xdata, ydata)), 5) == (
(-12986226192544605*x**4 +
8657484128363070*x**3 - 30301194449270745*x**2 + 4328742064181535*x
- 4328742064181535)/(x**3 + 9*x**2 - 3*x + 11))
def test_viete():
r1, r2 = symbols('r1, r2')
assert viete(
a*x**2 + b*x + c, [r1, r2], x) == [(r1 + r2, -b/a), (r1*r2, c/a)]
raises(ValueError, lambda: viete(1, [], x))
raises(ValueError, lambda: viete(x**2 + 1, [r1]))
raises(MultivariatePolynomialError, lambda: viete(x + y, [r1]))
|