
    /ii
                        d Z ddlZddlZddlmZ ddlmZmZm	Z	 ddl
mZ ej                            dg d          d             Zej                            dg d          d	             Zd
 Zd Zej                            dded         d         fded         d         fded         d         fg          d             Zd Zd ZdS )z(Tests for dataframe detection functions.    N)dependent_packages)is_df_or_seriesis_pandas_dfis_polars_df)_convert_containerconstructor_name)pyarrow	dataframepolarsc                     t          g dg dg|           }t          |          sJ t          t          j        g d                    rJ d S )N            r      r   r   r   )r   r   npasarrayr   dfs     h/var/www/html/bet.cuttalo.com/ml/venv/lib/python3.11/site-packages/sklearn/utils/tests/test_dataframe.pytest_is_df_or_seriesr      s[    	YYY			24D	E	EB2rz)))445555555    c                     t          g dg dg|           }| dv rt          |          rJ d S t          |          sJ d S )Nr   r   )r	   r   )r   r   r   s     r   !test_is_pandas_df_other_librariesr      s_    	YYY			24D	E	EB000#######Br   c                      t          j        d          } |                     g dg          }t          |          sJ t          t	          j        g d                    rJ t          d          rJ dS )z8Check behavior of is_pandas_df when pandas is installed.pandasr   r   N)pytestimportorskip	DataFramer   r   r   )pdr   s     r   test_is_pandas_dfr$      sw    		X	&	&B	yyyk	"	"BBJyyy1122222Ar   c                 r    t          t          j        g d                    rJ t          d          rJ dS )z0Check is_pandas_df when pandas is not installed.r   r   N)r   r   r   )hide_available_pandass    r   &test_is_pandas_df_pandas_not_installedr'   %   s>     BJyyy1122222Ar   zconstructor_name, minversionr	   r
   r   r   c                     t          g dg dg| |          }| dv rt          |          rJ d S t          |          sJ d S )Nr   r   )
minversion)r	   r
   )r   r   )r   r)   r   s      r   !test_is_polars_df_other_librariesr*   ,   sp     
	III
 
 
B
 333#######Br   c                  V     G d d          }  |             }t          |          rJ dS )z@Check is_polars_df for object that looks like a polars dataframec                       e Zd Zd ZdS )Ntest_is_polars_df_for_duck_typed_polars_dataframe.<locals>.NotAPolarsDataFramec                 &    g d| _         d| _        d S )Nr   	my_schemacolumnsschemaselfs    r   __init__zWtest_is_polars_df_for_duck_typed_polars_dataframe.<locals>.NotAPolarsDataFrame.__init__D   s    $99DL%DKKKr   N__name__
__module____qualname__r5    r   r   NotAPolarsDataFramer-   C   s#        	& 	& 	& 	& 	&r   r;   Nr   )r;   not_a_polars_dfs     r   1test_is_polars_df_for_duck_typed_polars_dataframer>   @   sT    & & & & & & & &
 *)++OO,,,,,,,r   c                  R     G d d          } t           |                       rJ dS )z?Check that is_polars_df return False for non-dataframe objects.c                       e Zd Zd ZdS )*test_is_polars_df.<locals>.LooksLikePolarsc                 *    ddg| _         ddg| _        d S )Nabr0   r3   s    r   r5   z3test_is_polars_df.<locals>.LooksLikePolars.__init__P   s    :DL*DKKKr   Nr6   r:   r   r   LooksLikePolarsrA   O   s#        	% 	% 	% 	% 	%r   rE   Nr<   )rE   s    r   test_is_polars_dfrF   L   sN    % % % % % % % %
 OO--.......r   )__doc__numpyr   r    sklearn._min_dependenciesr   sklearn.utils._dataframer   r   r   sklearn.utils._testingr   markparametrizer   r   r$   r'   r*   r>   rF   r:   r   r   <module>rN      s   . .      8 8 8 8 8 8 P P P P P P P P P P 5 5 5 5 5 5 +-O-O-OPP6 6 QP6 +-O-O-OPP    QP      "	&y1!45	(2156	%h/23 	  	  	 	- 	- 	-/ / / / /r   