
    sg                     ~    d dl mZ d dlmZmZmZmZmZmZm	Z	m
Z
 d dlmZmZmZ d dlmZmZ d dlmZmZ d Zd Zd Zy	)
    )raises)symbolsExprTupleIntegercossolveset	FiniteSetImageSet)_create_ranges_plot_sympifyextract_solution)ReferenceFrameVector)
CoordSys3Dr   c                  B   t        d      \  } }| |z   }t        |      }||k(  sJ | |z   df}t        |      }t        |t        t        t
        f      rt        |      dk(  sJ t        |d   t              sJ t        |d   t              sJ | |z   | ddfdddddf}t        |      }t        |t        t        t
        f      rt        |      dk(  sJ t        |d   t              sJ t        |d   t
              sJ t        |d   t              sJ t        |d	   t              sJ t        |d
   t              r,t        |d
   d   t              rt        |d
   d   t              sJ | |z   |ddfdf| dz   | ddfdff}t        |      }t        |t        t        t
        f      rt        |      dk(  sJ t        |d   t
              sJ t        |d   d   t
              sJ t        |d   d   d   t              sJ t        |d   d   t              sJ t        |d   t
              sJ t        |d   d   t
              sJ t        |d   d   d   t              sJ t        |d   d   t              sJ t        d      }d|j                  z  |j                  z   }t!        d      }d|j"                  z  |j$                  z   }||f}t        |      }t        |t        t        t
        f      rt        |      dk(  sJ t        |t&              sJ t        |t(              sJ y )Nx, y      r   strg       @)r   r            az$f_{1}$RC)r   r   
isinstancelisttupler   lenr   r   r   dictintfloatr   xyr   ij
MechVectorr   )r$   r%   argsrr   v1r   v2s           R/var/www/html/venv/lib/python3.12/site-packages/sympy/plotting/tests/test_utils.pytest_plot_sympifyr.   
   s   6?DAq q5DdA99 E1:DdAa$u-.3q6Q;>>adD!!!adG$$$ EAq!9eQA#7DdAa$u-.3q6Q;>>adD!!!adE"""adC   adG$$$adD!j1a#&>:aPQdSTgW\C]]] UQ1Is#a!eaAY	%BCDdAa$u-.3q6Q;>>adE"""ad1gu%%%ad1gaj'***ad1gs###adE"""ad1gu%%%ad1gaj'***ad1gs###
 	sA	
QSS133B3A	
QSS133B8DdAa$u-.3q6Q;>>b*%%%b&!!!    c                     t        d      \  t        hg d      } t        | t        t        t
        f      rt        |       dk(  sJ t        | d   t
        t        f      sJ | d   ddfk(  sJ t        hg d      } t        | t        t        t
        f      rt        |       dk(  sJ t        | d   t
        t        f      sJ t        | d   t
        t        f      sJ | d   ddfk(  sddfsJ | d   ddfk(  sddfsJ | d   | d   k7  sJ t        hddfgd      } t        | t        t        t
        f      rt        |       dk(  sJ t        | d   t
        t        f      sJ t        | d   t
        t        f      sJ | d   ddfk(  sddfsJ | d   ddfk(  sddfsJ | d   | d   k7  sJ t        t        fd       t        t        fd       y )	Nr   r   r   i
   r   c                  "    t         hg d      S )Nr   r   r$   r%   s   r-   <lambda>z$test_create_ranges.<locals>.<lambda>_   s    ~q!fb!< r/   c                  2    t         h ddfddfgd      S )Nr   r   r   r3   r4   s   r-   r5   z$test_create_ranges.<locals>.<lambda>`   s%    ~q!f1ay1a)6LaP r/   )	r   r   r   r   r   r   r    r   
ValueError)r*   r$   r%   s    @@r-   test_create_rangesr8   >   s   6?DAq 	sB"Aa$u-.3q6Q;>>adUEN+++Q4AsB<1vr1%Aa$u-.3q6Q;>>adUEN+++adUEN+++Q4AsB<AsB<//Q4AsB<AsB<//Q41Q4<< 		
A1I	
 	
	A a$u-.3q6Q;>>adUEN+++adUEN+++Q4Aq!9C,,Q4AsB<Aq!9,,Q41Q4<< :<=
:PQr/   c                  ,   t        d      } t        t        d| z              }|j                  t              sJ t        |      }t        |      dk(  sJ t        |t              sJ t        |d      }t        |      dk(  sJ t        |t              sJ y )Nr$   r1      (   )	r   r	   r   hasr   r   r    r   r
   )r$   solress      r-   test_extract_solutionr?   c   s    A
3rAv;
C778
3
Cs8r>>c9%%%
3
#Cs8r>>c9%%%r/   N)pytestr   sympyr   r   r   r   r   r	   r
   r   sympy.plotting.utilsr   r   r   sympy.physics.mechanicsr   r   r(   sympy.vectorr   r.   r8   r?    r/   r-   <module>rF      s<    G G G5 5 H +1"h"RJ&r/   