U
    KvfU                     @   s  d dl Zd dlmZ d dlmZ d dlZd dlmZ d dl	m
Z d dlmZ d dlmZmZmZmZ d dlmZ G dd dZG d	d
 d
eZG dd deZG dd deZG dd deZG dd deZG dd dZG dd deZG dd dZdd Zejj dd Z!G dd dZ"G dd  d Z#G d!d" d"Z$G d#d$ d$Z%d%d& Z&ejj ej'd'i d(d)d*gej'd+d,d-gej'd.d,d-gej'd/dd0d1d2d3gd4d5 Z(ejj d6d7 Z)dS )8    N)assert_equal)stats)gofplots)ProbPlotqqlineqqplotqqplot_2samples)_import_mplc                   @   s8  e Zd Zdd Zejjdd Zejjdd Zejjdd Z	ejjd	d
 Z
ejjdd Zejjdd Zejjddejjdd Zejjdd Zejjdd Zejjddejjdd Zejjdd Zejjdd Zejjdd Zejjdd  Zejjd!d" Zejjd#d$ Zd%d& Zd'S )(BaseProbplotMixinc                 C   sn   z dd l m} | \| _| _W n tk
r4   Y nX tjj| j	j
jd| _t| j| _tddddd| _d S )Nr   sizedcornflowerbluewhite      ?)markermarkerfacecolormarkeredgecoloralpha)matplotlib.pyplotpyplotsubplotsfigaxImportErrornprandomnormalprbpltdatashapeother_arrayr   other_prbplotdictplot_optionsselfplt r(   L/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/graphics/tests/test_gofplots.pysetup_method   s    zBaseProbplotMixin.setup_methodc                 C   s"   | j jf | j| jd| j d S Nr   liner   r   r   r-   r$   r&   close_figuresr(   r(   r)   test_qqplot#   s    zBaseProbplotMixin.test_qqplotc                 C   s   | j j| j| jd d S r+   r   ppplotr   r-   r/   r(   r(   r)   test_ppplot'   s    zBaseProbplotMixin.test_ppplotc                 C   s"   | j jf | j| jd| j d S r+   r   probplotr   r-   r$   r/   r(   r(   r)   test_probplot+   s    zBaseProbplotMixin.test_probplotc                 C   s$   | j jf | jd| jd| j d S )NT)r   Zexceedr-   r5   r/   r(   r(   r)   test_probplot_exceed/   s      z&BaseProbplotMixin.test_probplot_exceedc                 C   s&   | j jf | j| j| jd| j d S N)r   r-   other)r   r   r   r-   r!   r$   r/   r(   r(   r)   test_qqplot_other_array5   s    z)BaseProbplotMixin.test_qqplot_other_arrayc                 C   s&   | j jf | j| j| jd| j d S r9   )r   r3   r   r-   r!   r$   r/   r(   r(   r)   test_ppplot_other_array>   s    z)BaseProbplotMixin.test_ppplot_other_arrayT)strictc                 C   s&   | j jf | j| j| jd| j d S r9   )r   r6   r   r-   r!   r$   r/   r(   r(   r)   test_probplot_other_arrayG   s    z+BaseProbplotMixin.test_probplot_other_arrayc                 C   s&   | j jf | j| j| jd| j d S r9   )r   r   r   r-   r"   r$   r/   r(   r(   r)   test_qqplot_other_prbpltQ   s    z*BaseProbplotMixin.test_qqplot_other_prbpltc                 C   s&   | j jf | j| j| jd| j d S r9   )r   r3   r   r-   r"   r$   r/   r(   r(   r)   test_ppplot_other_prbpltZ   s    z*BaseProbplotMixin.test_ppplot_other_prbpltc                 C   s&   | j jf | j| j| jd| j d S r9   )r   r6   r   r-   r"   r$   r/   r(   r(   r)   test_probplot_other_prbpltc   s    z,BaseProbplotMixin.test_probplot_other_prbpltc                 C   s&   | j jf | j| jddd| j d S NzCustom X-LabelzCustom Y-Label)r   r-   xlabelylabelr.   r/   r(   r(   r)   test_qqplot_custom_labelsm   s    z+BaseProbplotMixin.test_qqplot_custom_labelsc                 C   s&   | j jf | j| jddd| j d S rB   )r   r3   r   r-   r$   r/   r(   r(   r)   test_ppplot_custom_labelsw   s    z+BaseProbplotMixin.test_ppplot_custom_labelsc                 C   s&   | j jf | j| jddd| j d S rB   r5   r/   r(   r(   r)   test_probplot_custom_labels   s    z-BaseProbplotMixin.test_probplot_custom_labelsc                 C   s    | j j| j| jddddd d S Nr   r   r   r   )r   r-   r   r   r   r   )r   r   r   r-   r/   r(   r(   r)   test_qqplot_pltkwargs   s    z'BaseProbplotMixin.test_qqplot_pltkwargsc                 C   s    | j j| j| jddddd d S rH   r2   r/   r(   r(   r)   test_ppplot_pltkwargs   s    z'BaseProbplotMixin.test_ppplot_pltkwargsc                 C   s    | j j| j| jddddd d S rH   )r   r6   r   r-   r/   r(   r(   r)   test_probplot_pltkwargs   s    z)BaseProbplotMixin.test_probplot_pltkwargsc                 C   s4   | j jd | j jkst| j jd | j jks0td S )N)r   
fit_paramslocAssertionErrorscaler&   r(   r(   r)   test_fit_params   s    z!BaseProbplotMixin.test_fit_paramsN)__name__
__module____qualname__r*   pytestmark
matplotlibr1   r4   r7   r8   r;   r<   Zxfailr>   r?   r@   rA   rE   rF   rG   rI   rJ   rK   rS   r(   r(   r(   r)   r
      sH   








	
	
	





r
   c                       s   e Zd Z fddZ  ZS )TestProbPlotLongelyNoFitc                    sv   t jd tjj | _tj| jj	dd| j_	t
| jj| jj	 | _t| jjtjddd| _d| _t   d S )N   Fprepend   distdistargsfitrr   r   seedsmdatasetslongleyloadr   add_constantexogOLSendogrc   mod_fitr   residr   tr   r-   superr*   rR   	__class__r(   r)   r*      s       z%TestProbPlotLongelyNoFit.setup_methodrT   rU   rV   r*   __classcell__r(   r(   rs   r)   rZ      s   rZ   c                       s   e Zd Z fddZ  ZS )TestProbPlotLongelyWithFitc                    sv   t jd tjj | _tj| jj	dd| j_	t
| jj| jj	 | _t| jjtjddd| _d| _t   d S )Nr[   Fr\   r^   Tr`   rd   re   rR   rs   r(   r)   r*      s       z'TestProbPlotLongelyWithFit.setup_methodru   r(   r(   rs   r)   rw      s   rw   c                       s   e Zd Z fddZ  ZS )TestProbPlotRandomNormalMinimalc                    sD   t jd t jjdddd| _t| j| _d | _tt	| 
  d S )Nr[         @      
@%   rO   rQ   r   )r   r   rf   r   r   r   r   r-   rr   rx   r*   rR   rs   r(   r)   r*      s
    z,TestProbPlotRandomNormalMinimal.setup_methodru   r(   r(   rs   r)   rx      s   rx   c                       s   e Zd Z fddZ  ZS )TestProbPlotRandomNormalWithFitc                    sH   t jd t jjdddd| _t| jdd| _d| _tt	| 
  d S )	Nr[   ry   rz   r{   r|   T)rc   q)r   r   rf   r   r   r   r   r-   rr   r}   r*   rR   rs   r(   r)   r*      s
    z,TestProbPlotRandomNormalWithFit.setup_methodru   r(   r(   rs   r)   r}      s   r}   c                       s4   e Zd Z fddZdd Zdd Zdd Z  ZS )	 TestProbPlotRandomNormalFullDistc                    sN   t jd t jjdddd| _t| jtjdddd	| _d
| _	t
   d S )Nr[   ry   rz   r{   r|         !@      @rO   rQ   ra   45)r   r   rf   r   r   r   r   normr   r-   rr   r*   rR   rs   r(   r)   r*      s
    z-TestProbPlotRandomNormalFullDist.setup_methodc                 C   s   | j jdkstd S )Nr   r   rO   rP   rR   r(   r(   r)   test_loc_set   s    z-TestProbPlotRandomNormalFullDist.test_loc_setc                 C   s   | j jdkstd S )Nr   r   rQ   rP   rR   r(   r(   r)   test_scale_set   s    z/TestProbPlotRandomNormalFullDist.test_scale_setc              	   C   s   t t  t| jtjddddd W 5 Q R X t t  t| jtjddddd W 5 Q R X t t  t| jtjddddd W 5 Q R X t t  t| jtjddddd	 W 5 Q R X d S )
Nr   r   r   T)ra   rc   )r   r   ra   rb   )ra   rO   )ra   rQ   )rW   raises
ValueErrorr   r   r   r   rR   r(   r(   r)   test_exceptions   s    $$z0TestProbPlotRandomNormalFullDist.test_exceptions)rT   rU   rV   r*   r   r   r   rv   r(   r(   rs   r)   r      s   r   c                   @   s4   e Zd Zdd Zejjdd Zejjdd ZdS )TestCompareSamplesDifferentSizec                 C   s@   t jd tt jjdddd| _tt jjdddd| _d S )Nr[   ry   rz   r{   r|   7   )r   r   rf   r   r   data1data2rR   r(   r(   r)   r*      s    z,TestCompareSamplesDifferentSize.setup_methodc              	   C   s:   | j j| jd tt | jj| j d W 5 Q R X d S N)r:   )r   r   r   rW   r   r   r/   r(   r(   r)   r1      s    z+TestCompareSamplesDifferentSize.test_qqplotc                 C   s$   | j j| jd | jj| j d d S r   )r   r3   r   r/   r(   r(   r)   r4     s    z+TestCompareSamplesDifferentSize.test_ppplotN)	rT   rU   rV   r*   rW   rX   rY   r1   r4   r(   r(   r(   r)   r      s
   
r   c                       s<   e Zd Z fddZdd Zdd Zdd Zd	d
 Z  ZS )$TestProbPlotRandomNormalLocScaleDistc                    sJ   t jd t jjdddd| _t| jddd| _d	| _tt	| 
  d S )
Nr[   ry   rz   r{   r|         r   r   )r   r   rf   r   r   r   r   r-   rr   r   r*   rR   rs   r(   r)   r*     s
    z1TestProbPlotRandomNormalLocScaleDist.setup_methodc                 C   s   | j jdkstd S )Nr   r   rR   r(   r(   r)   r     s    z1TestProbPlotRandomNormalLocScaleDist.test_loc_setc                 C   s   | j jdkstd S )Nr   r   rR   r(   r(   r)   r     s    z3TestProbPlotRandomNormalLocScaleDist.test_scale_setc                 C   s   | j j dkstd S )Ng       @)r   ra   ZmeanrP   rR   r(   r(   r)   test_loc_set_in_dist  s    z9TestProbPlotRandomNormalLocScaleDist.test_loc_set_in_distc                 C   s   | j j dkstd S )Ng      "@)r   ra   varrP   rR   r(   r(   r)   test_scale_set_in_dist  s    z;TestProbPlotRandomNormalLocScaleDist.test_scale_set_in_dist)	rT   rU   rV   r*   r   r   r   r   rv   r(   r(   rs   r)   r     s
   r   c                   @   sT   e Zd Zdd Zejjdd Zejjdd Zejjdd Z	ejjd	d
 Z
dS )TestTopLevelc                 C   s   t jj | _t j| jjdd| j_t | jj| jj	 | _
| j
j| _t| j
jtjdd| _tjj| jjjd| _t| j| _d S )NFr\   r^   r   r   )rg   rh   ri   rj   r   rk   rl   rm   rn   rc   ro   rp   resr   r   rq   r   r   r   r   r    r!   r"   rR   r(   r(   r)   r*   "  s    
zTestTopLevel.setup_methodc                 C   s   t | jdd d S Nrd   r-   r   r   r/   r(   r(   r)   r1   +  s    zTestTopLevel.test_qqplotc                 C   s   t | jdddddd d S )Nrd   r   r   r   r   )r-   r   r   r   r   r   r/   r(   r(   r)   rI   /  s    z"TestTopLevel.test_qqplot_pltkwargsc                 C   s    dD ]}t | j| j|d qd S N)rd   r~   r   sr   r   r   r"   r&   r0   r-   r(   r(   r)   &test_qqplot_2samples_prob_plot_objects:  s    z3TestTopLevel.test_qqplot_2samples_prob_plot_objectsc                 C   s    dD ]}t | j| j|d qd S r   r   r   r!   r   r(   r(   r)   test_qqplot_2samples_arraysA  s    z(TestTopLevel.test_qqplot_2samples_arraysN)rT   rU   rV   r*   rW   rX   rY   r1   rI   r   r   r(   r(   r(   r)   r   !  s   	



r   c              	   C   sn   t jd tjj }tj|jdd|_t	|j
|j }tjtdd t|jtjdd W 5 Q R X d S )Nr[   Fr\   zdist\(0, 1, 4, loc=0, scale=1\))match)r      r_   )rb   )r   r   rf   rg   rh   ri   rj   rk   rl   rm   rn   rc   rW   r   	TypeErrorr   rp   r   rq   )r0   r   ro   r(   r(   r)   test_invalid_dist_configI  s    r   c                  C   s   t jd} | d}| d}t||}t||}| d  d  \}}| d  d  \}}t j	|| t j	|| t
| d  }	t
| d  }
|	|
ksttjjdd }tjjdd }tjjdd	 }d S )
Nr   d      c                 S   s   t | jdd d S r   r   r/   r(   r(   r)   r1   b  s    z(test_qqplot_unequal.<locals>.test_qqplotc                 S   s    dD ]}t | j| j|d qd S r   r   r   r(   r(   r)   "test_qqplot_2samples_prob_plot_objf  s    z?test_qqplot_unequal.<locals>.test_qqplot_2samples_prob_plot_objc                 S   s    dD ]}t | j| j|d qd S r   r   r   r(   r(   r)   r   m  s    z8test_qqplot_unequal.<locals>.test_qqplot_2samples_arrays)r   r   RandomStateZstandard_normalr   get_axesget_childrenget_datatestingZassert_allcloselenrP   rW   rX   rY   )rsr   r   Zfig1Zfig2x1y1Zx2y2Znumobj1Znumobj2r1   r   r   r(   r(   r)   test_qqplot_unequalS  s$    





r   c                   @   s   e Zd Zdd Zdd ZdS )TestCheckDistc                 C   s    t tjd t tjd d S )Nppfcdf)r   
_check_forr   r   rR   r(   r(   r)   	test_goodv  s    zTestCheckDist.test_goodc              	   C   sH   t t tdd W 5 Q R X t t tdd W 5 Q R X d S )Njunkr   r   )rW   r   AttributeErrorr   r   rR   r(   r(   r)   test_badz  s    zTestCheckDist.test_badN)rT   rU   rV   r   r   r(   r(   r(   r)   r   u  s   r   c                   @   s   e Zd Zdd Zejjdd Zejjdd Zejjdd Z	ejjd	d
 Z
ejjdd Zejjdd Zejjdd ZdS )
TestDoPlotc              	   C   s   z dd l m} | \| _| _W n tk
r4   Y nX dddddddd	d
g	| _dddddddddg	| _dddddd| _ddd| _	d S )Nr   g?g333333?       @g      @g      $@g      I@g     T@gfffffX@gX@333333?gffffff?g333333?g @g	@g@gffffff@g333333@r   r   Z	firebrickg      ?z--)r   r   r   Zmarkeredgewidth	linestyle-Zmid)r   where)
r   r   r   r   r   r   xyfull_optionsstep_optionsr%   r(   r(   r)   r*     s    zTestDoPlot.setup_methodc                 C   sZ   t  }t| j| j\}}t||js*tt||js:t| j	|k	sHt| j
|k	sVtd S )N)r	   r   _do_plotr   r   
isinstanceFigurerP   Axesr   r   r&   r0   r'   r   r   r(   r(   r)   test_baseline  s    zTestDoPlot.test_baselinec                 C   s`   t  }tj| j| j| jd\}}t||js0tt||j	s@t| j
|ksNt| j|ks\td S )N)r   )r	   r   r   r   r   r   r   r   rP   r   r   r   r(   r(   r)   test_with_ax  s    zTestDoPlot.test_with_axc                 C   s&   t j| j| jf| jdd| j d S )NFr   stepr   r   r   r   r   r   r/   r(   r(   r)   test_plot_full_options  s    z!TestDoPlot.test_plot_full_optionsc                 C   s&   t j| j| jf| jdd| j d S NTr   )r   r   r   r   r   r   r/   r(   r(   r)   test_step_baseline  s    zTestDoPlot.test_step_baselinec                 C   s&   t j| j| jf| jdd| j d S r   r   r/   r(   r(   r)   test_step_full_options  s    z!TestDoPlot.test_step_full_optionsc                 C   s   t j| j| j| jdd d S )Nrd   r,   r   r   r   r   r   r/   r(   r(   r)   test_plot_qq_line  s    zTestDoPlot.test_plot_qq_linec                 C   s   t j| j| j| jddd d S )NTrd   )r   r   r-   r   r/   r(   r(   r)   test_step_qq_line  s    zTestDoPlot.test_step_qq_lineN)rT   rU   rV   r*   rW   rX   rY   r   r   r   r   r   r   r   r(   r(   r(   r)   r     s   


	
	
	
r   c                   @   s  e Zd Zdd Zejjdd Zejjdd Zejjdd Z	ejjd	d
 Z
ejjdd Zejjdd Zejjdd Zejjdd Zejjdd Zejjdd Zejjdd Zejjdd Zejjdd Zejjdd Zejjdd  Zejjd!d" Zd#S )$
TestQQLinec                 C   s   t jd t t jjdddd| _t t jjdddd| _z4dd lm} |	 \| _
| _| j| j| jd W n tk
r   Y nX d	d
dd| _d| _d S )Nr   g333333@r   r{   r|   r   g?ko   )
   r   r   r_   Zgreen)Z	linewidthZdashescolorzbo-)r   r   rf   sortr   r   r   r   r   r   r   r   Zplotr   lineoptionsfmtr%   r(   r(   r)   r*     s    zTestQQLine.setup_methodc              	   C   s&   t t t| jd W 5 Q R X d S )Nr   rW   r   r   r   r   rR   r(   r(   r)   test_badline  s    zTestQQLine.test_badlinec              	   C   s,   t t t| jd| jd W 5 Q R X d S )Nr   )r   )rW   r   r   r   r   r   r/   r(   r(   r)   test_non45_no_x  s    zTestQQLine.test_non45_no_xc              	   C   s,   t t t| jd| jd W 5 Q R X d S )Nr   )r   )rW   r   r   r   r   r   r/   r(   r(   r)   test_non45_no_y  s    zTestQQLine.test_non45_no_yc              	   C   s&   t t t| jd W 5 Q R X d S )Nr   r   r/   r(   r(   r)   test_non45_no_x_no_y  s    zTestQQLine.test_non45_no_x_no_yc                 C   s4   t | j }t| jd t | j |ks0td S )Nr   )r   r   r   r   rP   r&   r0   Z	nchildrenr(   r(   r)   test_45  s    zTestQQLine.test_45c                 C   s   t | jd| jd d S )Nr   )r   )r   r   r   r/   r(   r(   r)   test_45_fmt  s    zTestQQLine.test_45_fmtc                 C   s    t | jdfd| ji| j d S )Nr   r   )r   r   r   r   r/   r(   r(   r)   test_45_fmt_lineoptions   s    z"TestQQLine.test_45_fmt_lineoptionsc                 C   s>   t | j }t| jd| j| jd t | j |ks:td S )Nrd   r   r   r   r   r   r   r   r   rP   r   r(   r(   r)   test_r  s    zTestQQLine.test_rc                 C   s   t | jd| j| j| jd d S Nrd   r   r   r   r   r   r   r   r   r/   r(   r(   r)   
test_r_fmt
  s    zTestQQLine.test_r_fmtc                 C   s(   t | jdf| j| j| jd| j d S r   r   r   r   r   r   r   r/   r(   r(   r)   test_r_fmt_lineoptions  s       z!TestQQLine.test_r_fmt_lineoptionsc                 C   s>   t | j }t| jd| j| jd t | j |ks:td S )Nr   r   r   r   r(   r(   r)   test_s  s    zTestQQLine.test_sc                 C   s   t | jd| j| j| jd d S Nr   r   r   r/   r(   r(   r)   
test_s_fmt  s    zTestQQLine.test_s_fmtc                 C   s(   t | jdf| j| j| jd| j d S r   r   r/   r(   r(   r)   test_s_fmt_lineoptions  s       z!TestQQLine.test_s_fmt_lineoptionsc                 C   sB   t | j }t| jdtj| j| jd t | j |ks>td S )Nr~   )ra   r   r   )	r   r   r   r   r   r   r   r   rP   r   r(   r(   r)   test_q$  s    zTestQQLine.test_qc                 C   s"   t | jdtj| j| j| jd d S Nr~   )ra   r   r   r   )r   r   r   r   r   r   r   r/   r(   r(   r)   
test_q_fmt*  s    zTestQQLine.test_q_fmtc                 C   s,   t | jdftj| j| j| jd| j d S r   )r   r   r   r   r   r   r   r   r/   r(   r(   r)   test_q_fmt_lineoptions.  s    z!TestQQLine.test_q_fmt_lineoptionsN)rT   rU   rV   r*   rW   rX   rY   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r(   r(   r(   r)   r     sB   














r   c                   @   sl   e Zd Zdd Zdd Zejjdd Zejjdd Z	ejjd	d
 Z
ejjdd Zejjdd ZdS )TestPlottingPositionc                 C   s   d| _ t| j | _d S )N   )Nr   Zaranger   rR   r(   r(   r)   r*   <  s    z!TestPlottingPosition.setup_methodc                 C   s:   t j| j||d}tjj| j||d}tj||dd d S )Nab)r   betar[   )decimal)	r   Zplotting_posr   r   ZmstatsZplotting_positionsr   nptestZassert_array_almost_equal)r&   r   r  ZsmppZspppr(   r(   r)   do_test@  s      zTestPlottingPosition.do_testc                 C   s   |  dd d S )Nr   r  r/   r(   r(   r)   test_weibullH  s    z!TestPlottingPosition.test_weibullc                 C   s   |  dd d S )Nr   r   r  r/   r(   r(   r)   test_lininterpL  s    z#TestPlottingPosition.test_lininterpc                 C   s   |  dd d S )Nr   r  r/   r(   r(   r)   test_piecewiseP  s    z#TestPlottingPosition.test_piecewisec                 C   s   |  dd d S )NgUUUUUU?r  r/   r(   r(   r)   test_approx_med_unbiasedT  s    z-TestPlottingPosition.test_approx_med_unbiasedc                 C   s   |  dd d S )Ng?r  r/   r(   r(   r)   test_cunnaneX  s    z!TestPlottingPosition.test_cunnaneN)rT   rU   rV   r*   r  rW   rX   rY   r  r  r	  r
  r  r(   r(   r(   r)   r   ;  s   



r   c                  C   s  t ddddg} tt dtddd}t|j|  tt dtjddd}t|j|  tt dtjddd	}t|j|  t ddd
dg} tt dtddd
}t|j|  tt dtjddd
d}t|j|  t ddd
dg} tt dtddd
d}t|j|  tt dtjddd
dd}t|j|  tt dtjddd
dd}t|j|  tt dtjddd
dd}t|j|  tt dtjddd
dd}t|j|  d S )Nr   r   r   r   r   r   r   )r  r   r_   )r   r  rO   r[   )rQ   r   )r  rO   rQ   )r   r  rO   rQ   )r   arrayr   emptyr   r  r   rN   )expectedppr(   r(   r)   test_param_unpacking]  s.    r  labelsXY)rC   rD   x_size   2   y_sizer-   r   r   rd   r~   c                 C   s   t jd}|jdd|d}|jd|d}t|}	t|}
t|	|
fd|i|}| d }|	 }|
 }||k r|sd|kstd	|kstqd
|kstd|kstn6|sd	|kstd|kstnd|kstd
|kstd S )NiJ r   g?r|   r   r   r-   Z2ndZ1str  r  )r   r   r   r   Z
standard_trg   r   r   r   Z
get_xlabelZ
get_ylabelrP   )r0   Zreset_randomstater-   r  r  r  r   r   r   Zpp_xZpp_yr   r   Zx_labelZy_labelr(   r(   r)   test_correct_labelsy  s(    

r  c           	      C   s$  t jddd}t jddd}t||dd}| d }t | d }t | d }||k sjtt jddd}t||dd}| d }t | d }t | d }||k stt jddd}t||dd}| d }t | d }t | d }||k s td S )	Nr   r   )r   g{Gz?r   r   r   )i  )	r   r   r   r   r   ZdiffZget_ylimZget_xlimrP   )	r0   ZxxZxyr   r   Zy_rangeZx_rangeZxx_longZxy_longr(   r(   r)   test_axis_order  s&    r  )*Znumpyr   Znumpy.testingr   r  r   rW   Zscipyr   Zstatsmodels.apiapirg   Zstatsmodels.graphicsr   Zstatsmodels.graphics.gofplotsr   r   r   r   Zstatsmodels.graphics.utilsr	   r
   rZ   rw   rx   r}   r   r   r   r   r   rX   rY   r   r   r   r   r   r  Zparametrizer  r  r(   r(   r(   r)   <module>   sD     		(

!Mm"