Spaces:
Running
Running
File size: 3,127 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 |
from sympy.core.symbol import symbols, Symbol
from sympy.functions import Max
from sympy.plotting.experimental_lambdify import experimental_lambdify
from sympy.plotting.intervalmath.interval_arithmetic import \
interval, intervalMembership
# Tests for exception handling in experimental_lambdify
def test_experimental_lambify():
x = Symbol('x')
f = experimental_lambdify([x], Max(x, 5))
# XXX should f be tested? If f(2) is attempted, an
# error is raised because a complex produced during wrapping of the arg
# is being compared with an int.
assert Max(2, 5) == 5
assert Max(5, 7) == 7
x = Symbol('x-3')
f = experimental_lambdify([x], x + 1)
assert f(1) == 2
def test_composite_boolean_region():
x, y = symbols('x y')
r1 = (x - 1)**2 + y**2 < 2
r2 = (x + 1)**2 + y**2 < 2
f = experimental_lambdify((x, y), r1 & r2)
a = (interval(-0.1, 0.1), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(True, True)
a = (interval(-1.1, -0.9), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(False, True)
a = (interval(0.9, 1.1), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(False, True)
a = (interval(-0.1, 0.1), interval(1.9, 2.1))
assert f(*a) == intervalMembership(False, True)
f = experimental_lambdify((x, y), r1 | r2)
a = (interval(-0.1, 0.1), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(True, True)
a = (interval(-1.1, -0.9), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(True, True)
a = (interval(0.9, 1.1), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(True, True)
a = (interval(-0.1, 0.1), interval(1.9, 2.1))
assert f(*a) == intervalMembership(False, True)
f = experimental_lambdify((x, y), r1 & ~r2)
a = (interval(-0.1, 0.1), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(False, True)
a = (interval(-1.1, -0.9), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(False, True)
a = (interval(0.9, 1.1), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(True, True)
a = (interval(-0.1, 0.1), interval(1.9, 2.1))
assert f(*a) == intervalMembership(False, True)
f = experimental_lambdify((x, y), ~r1 & r2)
a = (interval(-0.1, 0.1), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(False, True)
a = (interval(-1.1, -0.9), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(True, True)
a = (interval(0.9, 1.1), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(False, True)
a = (interval(-0.1, 0.1), interval(1.9, 2.1))
assert f(*a) == intervalMembership(False, True)
f = experimental_lambdify((x, y), ~r1 & ~r2)
a = (interval(-0.1, 0.1), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(False, True)
a = (interval(-1.1, -0.9), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(False, True)
a = (interval(0.9, 1.1), interval(-0.1, 0.1))
assert f(*a) == intervalMembership(False, True)
a = (interval(-0.1, 0.1), interval(1.9, 2.1))
assert f(*a) == intervalMembership(True, True)
|