Ë
    êÿæi
  ã                   óf  — 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                  j                  dg d¢«      d„ «       Zej                  j                  dg d¢«      d	„ «       Zd
„ Zd„ Zej                  j                  dded   d   fded   d   fded   d   fg«      d„ «       Zd„ Zd„ Zy)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 ‚y )N©é   é   é   ©é   r   é   ©r   r   r   )r   r   ÚnpÚasarray©r   Údfs     úw/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/sklearn/utils/tests/test_dataframe.pyÚtest_is_df_or_seriesr      s<   € ä	šYª	Ð2Ð4DÓ	E€Bä˜2ÔÐÐÜœrŸz™zª)Ó4Ô5Ð5Ð5Ð5ó    c                 óf   — t        g d¢g d¢g| «      }| dv rt        |«      rJ ‚y t        |«      sJ ‚y )Nr   r   )r	   r   )r   r   r   s     r   Ú!test_is_pandas_df_other_librariesr      s<   € ä	šYª	Ð2Ð4DÓ	E€BØÐ0Ñ0Ü Ô#Ð#Ð#Ð#ä˜BÔÐÑr   c                  óÎ   — t        j                  d«      } | j                  g d¢g«      }t        |«      sJ ‚t        t	        j
                  g d¢«      «      rJ ‚t        d«      rJ ‚y)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$      sV   € ä	×	Ñ	˜XÓ	&€BØ	�‰’y�kÓ	"€BÜ˜ÔÐÐÜœBŸJ™J¢yÓ1Ô2Ð2Ð2Ü˜AŒÐÐˆr   c                 ób   — t        t        j                  g d¢«      «      rJ ‚t        d«      rJ ‚y)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*   € ô œBŸJ™J¢yÓ1Ô2Ð2Ð2Ü˜AŒÐÐˆr   zconstructor_name, minversionr	   r
   r   r   c                 ój   — t        g d¢g d¢g| |¬«      }| dv rt        |«      rJ ‚y t        |«      sJ ‚y )Nr   r   )Ú
minversion)r	   r
   )r   r   )r   r)   r   s      r   Ú!test_is_polars_df_other_librariesr*   ,   sF   € ô 
Ú	’IÐØØô
€Bð
 Ð3Ñ3Ü Ô#Ð#Ð#Ð#ä˜BÔÐÑr   c                  ó@   —  G d„ d«      }  | «       }t        |«      rJ ‚y)z@Check is_polars_df for object that looks like a polars dataframec                   ó   — e Zd Zd„ Zy)úNtest_is_polars_df_for_duck_typed_polars_dataframe.<locals>.NotAPolarsDataFramec                 ó$   — g d¢| _         d| _        y )Nr   Ú	my_schema©ÚcolumnsÚschema©Úselfs    r   Ú__init__zWtest_is_polars_df_for_duck_typed_polars_dataframe.<locals>.NotAPolarsDataFrame.__init__D   s   € Ú$ˆDŒLØ%ˆD�Kr   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>   @   s'   € ÷&ñ &ñ
 *Ó+€OÜ˜OÔ,Ð,Ð,Ð,r   c                  ó<   —  G d„ d«      } t         | «       «      rJ ‚y)z?Check that is_polars_df return False for non-dataframe objects.c                   ó   — e Zd Zd„ Zy)ú*test_is_polars_df.<locals>.LooksLikePolarsc                 ó(   — ddg| _         ddg| _        y )NÚaÚbr0   r3   s    r   r5   z3test_is_polars_df.<locals>.LooksLikePolars.__init__P   s   € Ø ˜:ˆDŒLØ ˜*ˆD�Kr   Nr6   r:   r   r   ÚLooksLikePolarsrA   O   s   „ ó	%r   rE   Nr<   )rE   s    r   Útest_is_polars_dfrF   L   s"   € ÷%ñ %ô
 ™OÓ-Ô.Ð.Ð.Ð.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ß PÑ PÝ 5ð ‡�×ÑÐ+Ò-OÓPñ6ó Qð6ð ‡�×ÑÐ+Ò-OÓPñ ó Qð òòð ‡�×ÑØ"à	Ð& yÑ1°!Ñ4Ð5Ø	Ð(¨Ñ2°1Ñ5Ð6Ø	Ð% hÑ/°Ñ2Ð3ðóñ	 óð	 ò	-ó/r   