M/Ph 1 d dl Zd dlmZ d dlmZ d dlmZ e Zej j
Z eej j
d Z ej eej d cZ Z
G d d Z G d d
Z G d d Z G d
d Z G d d Z G d d Z G d d Z G d d Z G d d Z G d d Z G d d Z G d d ZdS ) N) load_pandas)
OaxacaBlinder) add_constantF)prependc * e Zd Zed Zd ZdS )
TestOaxacac F t t t d | _ d S )N r endogexogmodelclss c/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/stats/tests/test_oaxaca.py setup_classzTestOaxaca.setup_class s !%q11 c t j d t j g d }t j g d }t j g d }| j j \ }}}}| j j \ }} }| j d j \ }
} } t j
||d d t j
||d d t j
||d d t j
||d d t j
||d d t j
| |d d t j
||d d t j
|
|d d t j
| |d d t j
| |d d d S ) Nr d]Fc@-t@g R@ rmr gClY`@gzNz;@)
Gh@g93Lm%P@gLBx@Tr
) nprandomseedarrayr
three_foldparamstwo_foldstdtestingassert_almost_equal)
self
stata_resultsstata_results_pooledstata_results_stdendowcoefintergapunexpexp endow_varcoef_var inter_vars
r test_resultszTestOaxaca.test_results s
q!J!J!JKK
!x(F(F(F G G H%H%H%H I I"&*"7"7"9"9"@ tUC *--//6sC)-)>)>t)D)D)H& 8Y
&&sM!,BBB
&&t]1-=qAAA
&&umA.>BBB
&&s,@,CQGGG
&&s,@,CQGGG
&&u.B1.EqIII
&&y2CA2FJJJ
&&x1B11EqIII
&&y2CA2FJJJJJr N__name__
__module__ __qualname__ classmethodr r4 r r r r sA 2 2 [2K K K K Kr r c * e Zd Zed Zd ZdS )TestOaxacaNoSwapc J t t t dd | _ d S )Nr
Fswapr r s r r z TestOaxacaNoSwap.setup_class3 s !%qu=== r c
t j g d }t j g d }| j j \ }}}}| j j \ }}}t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d d S N)d]FcgTg]C_d@r )rB gClY`gzNz;r r
r r r r r r! r" r# r% r& r' r( r) r+ r, r- r. r/ r0 s r r4 z TestOaxacaNoSwap.test_results7 L !L!L!LMM
!x(I(I(I J J "&*"7"7"9"9"@ tUC *--//6sC
&&sM!,BBB
&&t]1-=qAAA
&&umA.>BBB
&&s,@,CQGGG
&&s,@,CQGGG
&&u.B1.EqIIIIIr Nr5 r: r r r< r< 2 sA > > [> J J J J Jr r< c * e Zd Zed Zd ZdS )TestOaxacaPandasc F t t t d | _ d S )NOWNRENTr pd_endogpd_exogr r s r r z TestOaxacaPandas.setup_classG s !(GY?? r c
t j g d }t j g d }| j j \ }}}}| j j \ }}}t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d d S Nr r r r
r r rC rD s r r4 z TestOaxacaPandas.test_resultsK L !J!J!JKK
!x(F(F(F G G "&*"7"7"9"9"@ tUC *--//6sC
&&sM!,BBB
&&t]1-=qAAA
&&umA.>BBB
&&s,@,CQGGG
&&s,@,CQGGG
&&u.B1.EqIIIIIr Nr5 r: r r rG rG F sD @ @ [@ J J J J Jr rG c * e Zd Zed Zd ZdS )TestOaxacaPandasNoSwapc J t t t dd | _ d S )NrI Fr> rJ r s r r z"TestOaxacaPandasNoSwap.setup_class[ s !(GYUKKK r c
t j g d }t j g d }| j j \ }}}}| j j \ }}}t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d d S rA rC rD s r r4 z#TestOaxacaPandasNoSwap.test_results_ rE r Nr5 r: r r rQ rQ Z sD L L [L J J J J Jr rQ c * e Zd Zed Zd ZdS )TestOaxacaNoConstPassedc r t t j j t j j dd | _ d S )Nr
Fhasconstr pandas_dfr valuesr
r r s r r z#TestOaxacaNoConstPassed.setup_classo s/ ! O "IN$91u
r c
t j g d }t j g d }| j j \ }}}}| j j \ }}}t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d d S rN rC rD s r r4 z$TestOaxacaNoConstPassed.test_resultsu rO r Nr5 r: r r rU rU n A
[
J J J J Jr rU c * e Zd Zed Zd ZdS ) TestOaxacaNoSwapNoConstPassedc t t t j j t j j ddd | _ d S )Nr
FrX r? rY r s r r z)TestOaxacaNoSwapNoConstPassed.setup_class s5 ! O " N !
r c
t j g d }t j g d }| j j \ }}}}| j j \ }}}t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d d S rA rC rD s r r4 z*TestOaxacaNoSwapNoConstPassed.test_results rE r Nr5 r: r r r_ r_ A
[
J J J J Jr r_ c * e Zd Zed Zd ZdS ) TestOaxacaPandasNoConstPassedc ^ t t j t j dd | _ d S )NrI FrW r rZ r r
r r s r r z)TestOaxacaPandasNoConstPassed.setup_class s) ! OY^Y
r c
t j g d }t j g d }| j j \ }}}}| j j \ }}}t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d d S rN rC rD s r r4 z*TestOaxacaPandasNoConstPassed.test_results rO r Nr5 r: r r re re r] r re c * e Zd Zed Zd ZdS )#TestOaxacaPandasNoSwapNoConstPassedc ` t t j t j ddd | _ d S )NrI Fra rg r s r r z/TestOaxacaPandasNoSwapNoConstPassed.setup_class s/ ! O N
r c
t j g d }t j g d }| j j \ }}}}| j j \ }}}t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d d S rA rC rD s r r4 z0TestOaxacaPandasNoSwapNoConstPassed.test_results rE r Nr5 r: r r rj rj rc r rj c * e Zd Zed Zd ZdS ) TestOneModelc t j d t t j t j dd ddd | _ d S ) Nr rI FrW Tself_submittedr
two_fold_typesubmitted_weight) r r r r rZ r r
r# one_modelr s r r zTestOneModel.setup_class sT
q% OY^Y
(4'7!(
L
L
r c " | j j \ }}}| j j \ }}t j g d }t j ddg }t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d d S )N)g~k R@g~oӟT@3c@gA 3m%P@g/Q@r r
r r )rt r" r$ r r r% r& )r' r/ r0 r. unexp_stdexp_stdone_params_stata_resultsone_std_stata_resultss r r4 zTestOneModel.test_results s ./sC !^ / 7#%8,K,K,K#L#L "(H)= > >
&&u.Fq.I1MMM
&&s,DQ,GKKK
&&s,DQ,GKKK
&&y2G2JANNN
&&w0Ea0H!LLLLLr Nr5 r: r r rn rn sD M M [M M M M M Mr rn c * e Zd Zed Zd ZdS )
TestZeroModelc t j d t t j t j dd ddd | _ d S )Nr rI FrW Trp rq ) r r r r rZ r r
r#
zero_modelr s r r zTestZeroModel.setup_class sT
q& OY^Y
(4'7!(
L
L r c " | j j \ }}}| j j \ }}t j g d }t j ddg }t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d d S )N)g]C_dr rv g>yX5@r r r
r r )r~ r" r$ r r r% r& )r' r/ r0 r. rw rx zero_params_stata_resultszero_std_stata_resultss r r4 zTestZeroModel.test_results s /0sC !_ 0 7$&H-M-M-M$N$N!!#8Y*?!@!@
&&u.G.JANNN
&&s,Ea,H!LLL
&&s,Ea,H!LLL
&&y2H2KQOOO
&&w0Fq0I1MMMMMr Nr5 r: r r r| r| sD M M [M N N N N Nr r| c * e Zd Zed Zd ZdS )TestOmegaModelc t j d t t j t j dd dd | _ d S )Nr rI FrW Tnuemark)rr ) r r r r rZ r r
r# omega_modelr s r r zTestOmegaModel.setup_class sQ
q' OY^Y
(4y(
1
1 r c " | j j \ }}}| j j \ }}t j g d }t j ddg }t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d d S )N)g<1P3@g`9ga@rv g}͍M@gC H@r r
r r )r r" r$ r r r% r& )r' r/ r0 r. rw rx nue_params_stata_resultsnue_std_stata_resultss r r4 zTestOmegaModel.test_results s *1sC ! - 1 7#%8,L,L,L#M#M "(H)= > >
&&u.Fq.I1MMM
&&s,DQ,GKKK
&&s,DQ,GKKK
&&y2G2JANNN
&&w0Ea0H!LLLLLr Nr5 r: r r r r sA 2 2 [2 M M M M Mr r c * e Zd Zed Zd ZdS )TestPooledModelc t j d t t j t j dd d | _ d S )Nr rI FrW T) r r r r rZ r r
r# pooled_modelr s r r zTestPooledModel.setup_class sN
q( OY^Y
(4.. r c " | j j \ }}}| j j \ }}t j g d }t j ddg }t j ||d d t j ||d d t j ||d d t j ||d d t j ||d d d S )N)gvX;@g
Y`@rv g{)r s 7 7 7 7 7 7 2 2 2 2 2 2 0 0 0 0 0 0 KMM |IN)5999 O\\
NE& & & '
K K K K K K K K6J J J J J J J J(J J J J J J J J(J J J J J J J J(J J J J J J J J,J J J J J J J J4J J J J J J J J,J J J J J J J J4M M M M M M M M,N N N N N N N N,M M M M M M M M,N N N N N N N N N Nr