
    g                         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d	S )
    )raises)symbolsExprTupleIntegercossolveset	FiniteSetImageSet)_create_ranges_plot_sympifyextract_solution)ReferenceFrameVector)
CoordSys3Dr   c                     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                    rBt          |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 d S )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           [/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/plotting/tests/test_utils.pytest_plot_sympifyr.   
   s   6??DAq q5DdA9999 E1:DdAa$u-..>3q66Q;;;;adD!!!!!adG$$$$$ EAq!9eQA#7DdAa$u-..>3q66Q;;;;adD!!!!!adE"""""adC     adG$$$$$adD!!]j1a#&>&>]:aPQdSTgW\C]C]]]] UQ1Is#a!eaAY	%BCDdAa$u-..>3q66Q;;;;adE"""""ad1gu%%%%%ad1gaj'*****ad1gs#####adE"""""ad1gu%%%%%ad1gaj'*****ad1gs#####
 	sA	
QS13B3A	
QS13B8DdAa$u-..>3q66Q;;;;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         k    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         k    sJ t          t          fd           t          t          fd           d S )	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                  :    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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-..>3q66Q;;;;adUEN+++++Q4AsB<1vr1%%Aa$u-..>3q66Q;;;;adUEN+++++adUEN+++++Q4AsB<AsB<Q4AsB<AsB<Q41Q4<<<< 		
A1I	
 	
	 	A a$u-..>3q66Q;;;;adUEN+++++adUEN+++++Q4Aq!9CQ4AsB<Aq!9Q41Q4<<<< :<<<<<===
:PPPPPQQQQQr/   c                     t          d          } t          t          d| z                      }|                    t                    sJ t          |          }t          |          dk    sJ t          |t                    sJ t          |d          }t          |          dk    sJ t          |t                    sJ d S )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s88r>>>>c9%%%%%
3
#
#Cs88r>>>>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 G G G G G G G G G G G G G G G G G5 5 5 5 5 5 5 5 5 5 H H H H H H H H + + + + + + + +1" 1" 1"h"R "R "RJ& & & & &r/   