!g 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 Zy) N) load_pandas)
OaxacaBlinder) add_constantF)prependc " e Zd Zed Zd Zy)
TestOaxacac 8 t t t d | _ y )N r endogexogmodelclss `/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/stats/tests/test_oaxaca.py setup_classzTestOaxaca.setup_class s !%q1 c t j j d t j g d }t j g d }t j g d }| j j j \ }}}}| j j j \ }} }| j j d j \ }
} } t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j | |d d t j j ||d d t j j |
|d d t j j | |d d t j j | |d d y ) 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!JK
!xx(F G HH%H I"&**"7"7"9"@"@ tUC **--/66sC)-)>)>t)D)H)H& 8Y
&&sM!,B
&&t]1-=qA
&&umA.>B
&&s,@,CQG
&&s,@,CQG
&&u.B1.EqI
&&y2CA2FJ
&&x1B11EqI
&&y2CA2FJr N__name__
__module__ __qualname__ classmethodr r4 r r r r s 2 2Kr r c " e Zd Zed Zd Zy)TestOaxacaNoSwapc < t t t dd | _ y )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 j j \ }}}t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d y 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 !LM
!xx(I J "&**"7"7"9"@"@ tUC **--/66sC
&&sM!,B
&&t]1-=qA
&&umA.>B
&&s,@,CQG
&&s,@,CQG
&&u.B1.EqIr Nr5 r: r r r< r< 2 s > > Jr r< c " e Zd Zed Zd Zy)TestOaxacaPandasc 8 t t t d | _ y )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 j j \ }}}t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d y Nr r r r
r r rC rD s r r4 z TestOaxacaPandas.test_resultsK !JK
!xx(F G "&**"7"7"9"@"@ tUC **--/66sC
&&sM!,B
&&t]1-=qA
&&umA.>B
&&s,@,CQG
&&s,@,CQG
&&u.B1.EqIr Nr5 r: r r rG rG F s @ @ Jr rG c " e Zd Zed Zd Zy)TestOaxacaPandasNoSwapc < t t t dd | _ y )NrI Fr> rJ r s r r z"TestOaxacaPandasNoSwap.setup_class[ s !(GYUK r c t j g d }t j g d }| j j j \ }}}}| j j j \ }}}t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d y rA rC rD s r r4 z#TestOaxacaPandasNoSwap.test_results_ rE r Nr5 r: r r rQ rQ Z s L L Jr rQ c " e Zd Zed Zd Zy)TestOaxacaNoConstPassedc t t j j t j j dd | _ y )Nr
Fhasconstr pandas_dfr valuesr
r r s r r z#TestOaxacaNoConstPassed.setup_classo s+ ! OO " "INN$9$91u
r c t j g d }t j g d }| j j j \ }}}}| j j j \ }}}t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d y rN rC rD s r r4 z$TestOaxacaNoConstPassed.test_resultsu rO r Nr5 r: r r rU rU n
Jr rU c " e Zd Zed Zd Zy) TestOaxacaNoSwapNoConstPassedc t t j j t j j ddd | _ y )Nr
FrX r? rY r s r r z)TestOaxacaNoSwapNoConstPassed.setup_class s1 ! OO " " NN ! !
r c t j g d }t j g d }| j j j \ }}}}| j j j \ }}}t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d y rA rC rD s r r4 z*TestOaxacaNoSwapNoConstPassed.test_results rE r Nr5 r: r r r_ r_
Jr r_ c " e Zd Zed Zd Zy) TestOaxacaPandasNoConstPassedc d t t j t j dd | _ y )NrI FrW r rZ r r
r r s r r z)TestOaxacaPandasNoConstPassed.setup_class s ! OOY^^Y
r c t j g d }t j g d }| j j j \ }}}}| j j j \ }}}t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d y 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 Zy)#TestOaxacaPandasNoSwapNoConstPassedc f t t j t j ddd | _ y )NrI Fra rg r s r r z/TestOaxacaPandasNoSwapNoConstPassed.setup_class s% ! OO NN
r c t j g d }t j g d }| j j j \ }}}}| j j j \ }}}t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d t j j ||d d y 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 Zy) TestOneModelc t j j d t t j
t j dd j ddd | _ y ) 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 sC
q% OOY^^Y
(4'7!(
L
r c . | j j \ }}}| j j \ }}t j g d }t j ddg }t j
j
||d d t j
j
||d d t j
j
||d d t j
j
||d d t j
j
||d d y )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#%88,K#L "(H)= >
&&u.Fq.I1M
&&s,DQ,GK
&&s,DQ,GK
&&y2G2JAN
&&w0Ea0H!Lr Nr5 r: r r rn rn s M M Mr rn c " e Zd Zed Zd Zy)
TestZeroModelc t j j d t t j
t j dd j ddd | _ y )Nr rI FrW Trp rq ) r r r r rZ r r
r#
zero_modelr s r r zTestZeroModel.setup_class sC
q& OOY^^Y
(4'7!(
L r c . | j j \ }}}| j j \ }}t j g d }t j ddg }t j
j
||d d t j
j
||d d t j
j
||d d t j
j
||d d t j
j
||d d y )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 //00sC !__ 0 0 7$&HH-M$N!!#8Y*?!@
&&u.G.JAN
&&s,Ea,H!L
&&s,Ea,H!L
&&y2H2KQO
&&w0Fq0I1Mr Nr5 r: r r r| r| s M M Nr r| c " e Zd Zed Zd Zy)TestOmegaModelc t j j d t t j
t j dd j dd | _ y )Nr rI FrW Tnuemark)rr ) r r r r rZ r r
r# omega_modelr s r r zTestOmegaModel.setup_class s@
q' OOY^^Y
(4y(
1 r c . | j j \ }}}| j j \ }}t j g d }t j ddg }t j
j
||d d t j
j
||d d t j
j
||d d t j
j
||d d t j
j
||d d y )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 **11sC ! - - 1 1 7#%88,L#M "(H)= >
&&u.Fq.I1M
&&s,DQ,GK
&&s,DQ,GK
&&y2G2JAN
&&w0Ea0H!Lr Nr5 r: r r r r s 2 2 Mr r c " e Zd Zed Zd Zy)TestPooledModelc t j j d t t j
t j dd j d | _ y )Nr rI FrW T) r r r r rZ r r
r# pooled_modelr s r r zTestPooledModel.setup_class s<
q( OOY^^Y
(4. r c . | j j \ }}}| j j \ }}t j g d }t j ddg }t j
j
||d d t j
j
||d d t j
j
||d d t j
j
||d d t j
j
||d d y )N)gvX;@g
Y`@rv g{)r s 7 2 0 M INN))59 OO\
NNE& '
K K6J J(J J(J J(J J,J J4J J,J J4M M,N N,M M,N Nr