o
    ¢Äi™  ã                   @   sJ   d dl Zd dlZd dlmZmZ ej dejefej	efg¡dd„ ƒZ
dS )é    N)ÚWeightVector32ÚWeightVector64zdtype, WeightVectorc                 C   sj   t j d¡ | ¡}t j d¡ | ¡}|||ƒ}t  |j¡jt  | ¡u s%J ‚t  |j¡jt  | ¡u s3J ‚dS )z0Check the `dtype` consistency of `WeightVector`.éd   N)ÚnpÚrandomZrandZastypeZasarrayÚwÚdtypeZaw)r   ZWeightVectorÚweightsZaverage_weightsZweight_vector© r
   ú`/home/kim/smarthome/.venv/lib/python3.10/site-packages/sklearn/utils/tests/test_weight_vector.pyÚtest_type_invariance
   s
   	
 r   )Únumpyr   ZpytestZsklearn.utils._weight_vectorr   r   ÚmarkZparametrizeZfloat32Zfloat64r   r
   r
   r
   r   Ú<module>   s    þþ