U md@sxddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z GdddZdd ZGd d d ZdS) N) assert_equal) summary_col) add_constant)OLSc@s<eZdZddZddZddZddZd d Zd d Zd S)TestSummaryLatexcCsxd}dddddg}t|}dddddg}d dd d dg}t||}t||}t||g}d |}t||dS Na \begin{table} \caption{} \label{} \begin{center} \begin{tabular}{lll} \hline & y I & y II \\ \hline const & 7.7500 & 12.4231 \\ & (1.1058) & (3.1872) \\ x1 & -0.7500 & -1.5769 \\ & (0.2368) & (0.6826) \\ R-squared & 0.7697 & 0.6401 \\ R-squared Adj. & 0.6930 & 0.5202 \\ \hline \end{tabular} \end{center} \end{table} \bigskip Standard errors in parentheses. r z %s )rrfitras_latexrselfdesiredxy1y2reg1reg2actualr^/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/iolib/tests/test_summary2.pytest_summarycolsz TestSummaryLatex.test_summarycolc Csd}dddddg}t|}dddddg}d dd d dg}t||}t||}t||gd d }d|}t||t||gdd d}dt|kstdt|kstdt|kstdS)Na/ ========================== y I y II -------------------------- const 7.7 12.4 (1.1) (3.2) x1 -0.7 -1.6 (0.2) (0.7) R-squared 0.8 0.6 R-squared Adj. 0.7 0.5 ========================== Standard errors in parentheses. rr r r r r rrrrz%0.1f float_formatz%s T)starsr z7.7***z12.4**z12.4***)rrrrZas_textrstrAssertionError) rrrrrrrrZstarredrrrtest_summarycol_float_format1s z-TestSummaryLatex.test_summarycol_float_formatc Csdddddg}t|}t|tdgdgdgdgdgggd}d dd ddg}d ddd dg}t||}t||}t||gd dgdd}dt|kstt||gdgdd}d t|kstdt|kstdS)Nrr r r r rr rrrconstx1T)regressor_orderZ drop_omittedx2F) rnp concatenatearrayrrrr"r#)rrr*rrrrrrrrtest_summarycol_drop_omittedPs * z-TestSummaryLatex.test_summarycol_drop_omittedc Cs dddddg}t|}t|tdgdgdgdgdgggd}tj|d d d gd }d ddddg}dddddg}t||}t||}ddddd}t||gdd}} t||gd d gd|d} d} t | dD](} | D]} | | r| t | kst qqdS)Nrr r r r%r&r rr'ba)columnsr rrrcSsdt|jS)Nz{:.3f})formatintZrsquaredrrrrlzFTestSummaryLatex.test_summary_col_ordering_preserved..cSsdt|jS)Nz{0:d})r2r3Znobsr4rrrr5mr6)ZR2Nz%0.4fr)r)r info_dict)r'r/r0 ) rr+r,r-pdZ DataFramerrrr"split startswithr#) rrr*rrrrr8originalr variableslinevariablerrr#test_summary_col_ordering_preservedas** z4TestSummaryLatex.test_summary_col_ordering_preservedcCsxd}dddddg}t|}dddddg}d dd d dg}t||}t||}t||g}d |}t||dSr)rrrrZ _repr_latex_rrrrrtest__repr_latex_xsz"TestSummaryLatex.test__repr_latex_c Csxdddddg}t|}dddddg}t||}ttd|}W5QRXd }||k}|d ksttdS) Nrr r r r r rignorez\end{tabular} \begin{tabular}T) rrrwarningscatch_warnings simplefiltersummaryrr#)rrrrrZstring_to_findresultrrrtest_OLSsummarys  z TestSummaryLatex.test_OLSsummaryN) __name__ __module__ __qualname__rr$r.rArBrIrrrrr s ""rc Csddddddddg}ddddddd dg}t|t|}d }tt|t|ks^tW5QRXtt|t| kstW5QRXt||d d }d }tt|t|kstW5QRXtt|t| kstW5QRXdS)Nrr r r rr r r% z R-squared:F)hasconstzR-squared (uncentered):) rrrpytestZwarns UserWarningr"Zsummary2r#rG)ryZreg_with_constantZr2_strZreg_without_constantrrrtest_ols_summary_rsquared_labels    rRc@s,eZdZdZeddZddZddZdS) TestSummaryLabelsz@ Test that the labels are correctly set in the summary tablecCs:ddddgd}tddddgd}t||d|_dS)Nrr rr )ZendogZexog)rrrmod)clsrQrrrr setup_classszTestSummaryLabels.setup_classcCs2t|jdd}dt|kstdt|ks.tdS)NTresultsZ include_r2z R-squared R-squared Adj.rrTr"r#rtablerrrtest_summary_col_r2sz%TestSummaryLabels.test_summary_col_r2cCs2t|jdd}dt|kstdt|ks.tdS)NFrWz R-squaredrYrZr[rrrtest_absence_of_r2sz$TestSummaryLabels.test_absence_of_r2N)rJrKrL__doc__ classmethodrVr]r^rrrrrSs  rS)rDnumpyr+Zpandasr:rOZ numpy.testingrZstatsmodels.iolib.summary2rZstatsmodels.tools.toolsrZ#statsmodels.regression.linear_modelrrrRrSrrrrs