U
    Kvfa                     @   sB   d dl Z d dlmZ d dlmZ d dlmZmZ G dd dZdS )    N)array)OLS)_generate_urlwebdocc                   @   s4   e Zd ZdZdZdd Zdd Zdd Zd	d
 ZdS )TestWeb#https://www.statsmodels.org/stable/"https://www.statsmodels.org/devel/c                 C   sX   t dd}|| jd kstt dd}|| jd ks8tt dd}|| jd ksTtd S )NarchTz2search.html?q=arch&check_keywords=yes&area=defaultFzdickey fullerz;search.html?q=dickey+fuller&check_keywords=yes&area=default)r   stableAssertionErrordevelselfurl r   D/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/tools/tests/test_web.pytest_string   s    


zTestWeb.test_stringc                 C   s<   t td}|| jd kstt td}|| jd ks8td S )NTz6generated/statsmodels.regression.linear_model.OLS.htmlF)r   r   r
   r   r   r   r   r   r   test_function   s    


zTestWeb.test_functionc                 C   s0   t d d}|dkstt d d}|dks,td S )NTr   Fr   )r   r   r   r   r   r   test_nothing"   s    

zTestWeb.test_nothingc              	   C   sD   t t ttd W 5 Q R X t t tdd W 5 Q R X d S )NT   F)pytestZraises
ValueErrorr   r   )r   r   r   r   test_errors(   s    zTestWeb.test_errorsN)	__name__
__module____qualname__r
   r   r   r   r   r   r   r   r   r   r      s   
r   )	r   Znumpyr   Z#statsmodels.regression.linear_modelr   Zstatsmodels.tools.webr   r   r   r   r   r   r   <module>   s   