o
    tBh                  	   @   s   d dl Z d dlZd dlZd dlZd dlmZ d dlm	Z	 d dlm
Z
 ejdedgedgedgfdd	 Zejd
dd dedfdgdd ZdS )    N)assert_array_equal)_object_dtype_isnan)
loguniformz
dtype, val   ac                 C   sJ   t j|t jgt j|gg| d}t ddgddgg}t|}t|| d S )N)dtypeFT)nparraynanr   r   )r   valXexpected_maskmask r   u/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/sklearn/utils/tests/test_fixes.pytest_object_dtype_isnan   s   r   zlow,high,base)r   
      )r   r   r   c                    s   t  |   | }t|tjjjsJ |jddd} |  |k r*| | k s,J t|dks4J t	
 fdd|D }t	|\}}| dksOJ t	||   dks^J t  |   | jddt  |   | jddkszJ d S )	Ni  r   )sizerandom_statec                    s   g | ]}t | qS r   )mathlog).0xbaser   r   
<listcomp>(   s    z#test_loguniform.<locals>.<listcomp>   (   )r   )r   
isinstancescipystats_distn_infrastructure	rv_frozenrvsalllenr   r	   	histogrammeanabsmax)lowhighr   rvr%   log_rvscounts_r   r   r   test_loguniform   s   $r2   )r   numpyr   pytestscipy.statsr!   sklearn.utils._testingr   sklearn.utils.fixesr   r   markparametrizeobjectfloatr   expr2   r   r   r   r   <module>   s   

