91 lines
3.9 KiB
Python
91 lines
3.9 KiB
Python
|
from sympy.core.relational import Eq
|
||
|
|
from sympy.core.singleton import S
|
||
|
|
from sympy.abc import x, y, z, s, t
|
||
|
|
from sympy.sets import FiniteSet, EmptySet
|
||
|
|
from sympy.geometry import Point
|
||
|
|
from sympy.vector import ImplicitRegion
|
||
|
|
from sympy.testing.pytest import raises
|
||
|
|
|
||
|
|
|
||
|
|
def test_ImplicitRegion():
|
||
|
|
ellipse = ImplicitRegion((x, y), (x**2/4 + y**2/16 - 1))
|
||
|
|
assert ellipse.equation == x**2/4 + y**2/16 - 1
|
||
|
|
assert ellipse.variables == (x, y)
|
||
|
|
assert ellipse.degree == 2
|
||
|
|
r = ImplicitRegion((x, y, z), Eq(x**4 + y**2 - x*y, 6))
|
||
|
|
assert r.equation == x**4 + y**2 - x*y - 6
|
||
|
|
assert r.variables == (x, y, z)
|
||
|
|
assert r.degree == 4
|
||
|
|
|
||
|
|
|
||
|
|
def test_regular_point():
|
||
|
|
r1 = ImplicitRegion((x,), x**2 - 16)
|
||
|
|
assert r1.regular_point() == (-4,)
|
||
|
|
c1 = ImplicitRegion((x, y), x**2 + y**2 - 4)
|
||
|
|
assert c1.regular_point() == (0, -2)
|
||
|
|
c2 = ImplicitRegion((x, y), (x - S(5)/2)**2 + y**2 - (S(1)/4)**2)
|
||
|
|
assert c2.regular_point() == (S(5)/2, -S(1)/4)
|
||
|
|
c3 = ImplicitRegion((x, y), (y - 5)**2 - 16*(x - 5))
|
||
|
|
assert c3.regular_point() == (5, 5)
|
||
|
|
r2 = ImplicitRegion((x, y), x**2 - 4*x*y - 3*y**2 + 4*x + 8*y - 5)
|
||
|
|
assert r2.regular_point() == (S(4)/7, S(9)/7)
|
||
|
|
r3 = ImplicitRegion((x, y), x**2 - 2*x*y + 3*y**2 - 2*x - 5*y + 3/2)
|
||
|
|
raises(ValueError, lambda: r3.regular_point())
|
||
|
|
|
||
|
|
|
||
|
|
def test_singular_points_and_multiplicty():
|
||
|
|
r1 = ImplicitRegion((x, y, z), Eq(x + y + z, 0))
|
||
|
|
assert r1.singular_points() == EmptySet
|
||
|
|
r2 = ImplicitRegion((x, y, z), x*y*z + y**4 -x**2*z**2)
|
||
|
|
assert r2.singular_points() == FiniteSet((0, 0, z), (x, 0, 0))
|
||
|
|
assert r2.multiplicity((0, 0, 0)) == 3
|
||
|
|
assert r2.multiplicity((0, 0, 6)) == 2
|
||
|
|
r3 = ImplicitRegion((x, y, z), z**2 - x**2 - y**2)
|
||
|
|
assert r3.singular_points() == FiniteSet((0, 0, 0))
|
||
|
|
assert r3.multiplicity((0, 0, 0)) == 2
|
||
|
|
r4 = ImplicitRegion((x, y), x**2 + y**2 - 2*x)
|
||
|
|
assert r4.singular_points() == EmptySet
|
||
|
|
assert r4.multiplicity(Point(1, 3)) == 0
|
||
|
|
|
||
|
|
|
||
|
|
def test_rational_parametrization():
|
||
|
|
p = ImplicitRegion((x,), x - 2)
|
||
|
|
assert p.rational_parametrization() == (x - 2,)
|
||
|
|
|
||
|
|
line = ImplicitRegion((x, y), Eq(y, 3*x + 2))
|
||
|
|
assert line.rational_parametrization() == (x, 3*x + 2)
|
||
|
|
|
||
|
|
circle1 = ImplicitRegion((x, y), (x-2)**2 + (y+3)**2 - 4)
|
||
|
|
assert circle1.rational_parametrization(parameters=t) == (4*t/(t**2 + 1) + 2, 4*t**2/(t**2 + 1) - 5)
|
||
|
|
circle2 = ImplicitRegion((x, y), (x - S.Half)**2 + y**2 - (S(1)/2)**2)
|
||
|
|
|
||
|
|
assert circle2.rational_parametrization(parameters=t) == (t/(t**2 + 1) + S(1)/2, t**2/(t**2 + 1) - S(1)/2)
|
||
|
|
circle3 = ImplicitRegion((x, y), Eq(x**2 + y**2, 2*x))
|
||
|
|
assert circle3.rational_parametrization(parameters=(t,)) == (2*t/(t**2 + 1) + 1, 2*t**2/(t**2 + 1) - 1)
|
||
|
|
|
||
|
|
parabola = ImplicitRegion((x, y), (y - 3)**2 - 4*(x + 6))
|
||
|
|
assert parabola.rational_parametrization(t) == (-6 + 4/t**2, 3 + 4/t)
|
||
|
|
|
||
|
|
rect_hyperbola = ImplicitRegion((x, y), x*y - 1)
|
||
|
|
assert rect_hyperbola.rational_parametrization(t) == (-1 + (t + 1)/t, t)
|
||
|
|
|
||
|
|
cubic_curve = ImplicitRegion((x, y), x**3 + x**2 - y**2)
|
||
|
|
assert cubic_curve.rational_parametrization(parameters=(t)) == (t**2 - 1, t*(t**2 - 1))
|
||
|
|
cuspidal = ImplicitRegion((x, y), (x**3 - y**2))
|
||
|
|
assert cuspidal.rational_parametrization(t) == (t**2, t**3)
|
||
|
|
|
||
|
|
I = ImplicitRegion((x, y), x**3 + x**2 - y**2)
|
||
|
|
assert I.rational_parametrization(t) == (t**2 - 1, t*(t**2 - 1))
|
||
|
|
|
||
|
|
sphere = ImplicitRegion((x, y, z), Eq(x**2 + y**2 + z**2, 2*x))
|
||
|
|
assert sphere.rational_parametrization(parameters=(s, t)) == (2/(s**2 + t**2 + 1), 2*t/(s**2 + t**2 + 1), 2*s/(s**2 + t**2 + 1))
|
||
|
|
|
||
|
|
conic = ImplicitRegion((x, y), Eq(x**2 + 4*x*y + 3*y**2 + x - y + 10, 0))
|
||
|
|
assert conic.rational_parametrization(t) == (
|
||
|
|
S(17)/2 + 4/(3*t**2 + 4*t + 1), 4*t/(3*t**2 + 4*t + 1) - S(11)/2)
|
||
|
|
|
||
|
|
r1 = ImplicitRegion((x, y), y**2 - x**3 + x)
|
||
|
|
raises(NotImplementedError, lambda: r1.rational_parametrization())
|
||
|
|
r2 = ImplicitRegion((x, y), y**2 - x**3 - x**2 + 1)
|
||
|
|
raises(NotImplementedError, lambda: r2.rational_parametrization())
|