| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110 |
- from pytest import raises
- from sympy import (
- symbols, Expr, Tuple, Integer, cos, solveset, FiniteSet, ImageSet)
- from sympy.plotting.utils import (
- _create_ranges, _plot_sympify, extract_solution)
- from sympy.physics.mechanics import ReferenceFrame, Vector as MechVector
- from sympy.vector import CoordSys3D, Vector
- def test_plot_sympify():
- x, y = symbols("x, y")
- # argument is already sympified
- args = x + y
- r = _plot_sympify(args)
- assert r == args
- # one argument needs to be sympified
- args = (x + y, 1)
- r = _plot_sympify(args)
- assert isinstance(r, (list, tuple, Tuple)) and len(r) == 2
- assert isinstance(r[0], Expr)
- assert isinstance(r[1], Integer)
- # string and dict should not be sympified
- args = (x + y, (x, 0, 1), "str", 1, {1: 1, 2: 2.0})
- r = _plot_sympify(args)
- assert isinstance(r, (list, tuple, Tuple)) and len(r) == 5
- assert isinstance(r[0], Expr)
- assert isinstance(r[1], Tuple)
- assert isinstance(r[2], str)
- assert isinstance(r[3], Integer)
- assert isinstance(r[4], dict) and isinstance(r[4][1], int) and isinstance(r[4][2], float)
- # nested arguments containing strings
- args = ((x + y, (y, 0, 1), "a"), (x + 1, (x, 0, 1), "$f_{1}$"))
- r = _plot_sympify(args)
- assert isinstance(r, (list, tuple, Tuple)) and len(r) == 2
- assert isinstance(r[0], Tuple)
- assert isinstance(r[0][1], Tuple)
- assert isinstance(r[0][1][1], Integer)
- assert isinstance(r[0][2], str)
- assert isinstance(r[1], Tuple)
- assert isinstance(r[1][1], Tuple)
- assert isinstance(r[1][1][1], Integer)
- assert isinstance(r[1][2], str)
- # vectors from sympy.physics.vectors module are not sympified
- # vectors from sympy.vectors are sympified
- # in both cases, no error should be raised
- R = ReferenceFrame("R")
- v1 = 2 * R.x + R.y
- C = CoordSys3D("C")
- v2 = 2 * C.i + C.j
- args = (v1, v2)
- r = _plot_sympify(args)
- assert isinstance(r, (list, tuple, Tuple)) and len(r) == 2
- assert isinstance(v1, MechVector)
- assert isinstance(v2, Vector)
- def test_create_ranges():
- x, y = symbols("x, y")
- # user don't provide any range -> return a default range
- r = _create_ranges({x}, [], 1)
- assert isinstance(r, (list, tuple, Tuple)) and len(r) == 1
- assert isinstance(r[0], (Tuple, tuple))
- assert r[0] == (x, -10, 10)
- r = _create_ranges({x, y}, [], 2)
- assert isinstance(r, (list, tuple, Tuple)) and len(r) == 2
- assert isinstance(r[0], (Tuple, tuple))
- assert isinstance(r[1], (Tuple, tuple))
- assert r[0] == (x, -10, 10) or (y, -10, 10)
- assert r[1] == (y, -10, 10) or (x, -10, 10)
- assert r[0] != r[1]
- # not enough ranges provided by the user -> create default ranges
- r = _create_ranges(
- {x, y},
- [
- (x, 0, 1),
- ],
- 2,
- )
- assert isinstance(r, (list, tuple, Tuple)) and len(r) == 2
- assert isinstance(r[0], (Tuple, tuple))
- assert isinstance(r[1], (Tuple, tuple))
- assert r[0] == (x, 0, 1) or (y, -10, 10)
- assert r[1] == (y, -10, 10) or (x, 0, 1)
- assert r[0] != r[1]
- # too many free symbols
- raises(ValueError, lambda: _create_ranges({x, y}, [], 1))
- raises(ValueError, lambda: _create_ranges({x, y}, [(x, 0, 5), (y, 0, 1)], 1))
- def test_extract_solution():
- x = symbols("x")
- sol = solveset(cos(10 * x))
- assert sol.has(ImageSet)
- res = extract_solution(sol)
- assert len(res) == 20
- assert isinstance(res, FiniteSet)
- res = extract_solution(sol, 20)
- assert len(res) == 40
- assert isinstance(res, FiniteSet)
|
