U Kv f@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_latexrselfdesiredxy1y2reg1reg2actualrI/tmp/pip-unpacked-wheel-2v6byqio/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||gdd d}dt|kstdt|kstdt|kstt||dS)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)Zstarsr z7.7***z12.4**z12.4***)rrrrZas_textstrAssertionErrorr) rrrrrrrrZstarredrrrtest_summarycol_float_format1sz-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_omittedQs * 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})formatintZrsquaredrrrrmzFTestSummaryLatex.test_summary_col_ordering_preserved..cSsdt|jS)Nz{0:d})r1r2Znobsr3rrrr4nr5)ZR2Nz%0.4fr)r(r info_dict)r&r.r/ ) rr*r+r,pdZ DataFramerrrr!split startswithr") rrr)rrrrr7originalr variableslinevariablerrr#test_summary_col_ordering_preservedbs** 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_ysz"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-r@rArHrrrrr 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"rF)ryZreg_with_constantZr2_strZreg_without_constantrrrtest_ols_summary_rsquared_labels    rQc@s,eZdZdZeddZddZddZdS) TestSummaryLabelsz@ Test that the labels are correctly set in the summary tablecCs:ddddgd}tddddgd}t||d|_dS)Nrr rr )ZendogZexog)rrrmod)clsrPrrrr setup_classszTestSummaryLabels.setup_classcCs2t|jdd}dt|kstdt|ks.tdS)NTresultsZ include_r2z R-squared R-squared Adj.rrSr!r"rtablerrrtest_summary_col_r2sz%TestSummaryLabels.test_summary_col_r2cCs2t|jdd}dt|kstdt|ks.tdS)NFrVz R-squaredrXrYrZrrrtest_absence_of_r2sz$TestSummaryLabels.test_absence_of_r2N)rIrJrK__doc__ classmethodrUr\r]rrrrrRs  rR)rCZnumpyr*Zpandasr9rNZ numpy.testingrZstatsmodels.iolib.summary2rZstatsmodels.tools.toolsrZ#statsmodels.regression.linear_modelrrrQrRrrrrs