<s> """<STR_LIT>""" <EOL> from statsmodels . compat . python import zip <EOL> import numpy as np <EOL> from numpy . testing import assert_equal , assert_almost_equal , assert_ <EOL> from statsmodels . tools . eval_measures import ( <EOL> maxabs , meanabs , medianabs , medianbias , mse , rmse , stde , vare , <EOL> aic , aic_sigma , aicc , aicc_sigma , bias , bic , bic_sigma , <EOL> hqic , hqic_sigma , iqr ) <EOL> def test_eval_measures ( ) : <EOL> x = np . arange ( <NUM_LIT:20> ) . reshape ( <NUM_LIT:4> , <NUM_LIT:5> ) <EOL> y = np . ones ( ( <NUM_LIT:4> , <NUM_LIT:5> ) ) <EOL> assert_equal ( iqr ( x , y ) , <NUM_LIT:5> * np . ones ( <NUM_LIT:5> ) ) <EOL> assert_equal ( iqr ( x , y , axis = <NUM_LIT:1> ) , <NUM_LIT:2> * np . ones ( <NUM_LIT:4> ) ) <EOL> assert_equal ( iqr ( x , y , axis = None ) , <NUM_LIT:9> ) <EOL> assert_equal ( mse ( x , y ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( mse ( x , y , axis = <NUM_LIT:1> ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_almost_equal ( rmse ( x , y ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <EOL> <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_almost_equal ( rmse ( x , y , axis = <NUM_LIT:1> ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <EOL> <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( maxabs ( x , y ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( maxabs ( x , y , axis = <NUM_LIT:1> ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( meanabs ( x , y ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( meanabs ( x , y , axis = <NUM_LIT:1> ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( meanabs ( x , y , axis = <NUM_LIT:0> ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( medianabs ( x , y ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( medianabs ( x , y , axis = <NUM_LIT:1> ) , <EOL> np . array ( [ <NUM_LIT:1.> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( bias ( x , y ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( bias ( x , y , axis = <NUM_LIT:1> ) , <EOL> np . array ( [ <NUM_LIT:1.> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( medianbias ( x , y ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( medianbias ( x , y , axis = <NUM_LIT:1> ) , <EOL> np . array ( [ <NUM_LIT:1.> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( vare ( x , y ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> assert_equal ( vare ( x , y , axis = <NUM_LIT:1> ) , <EOL> np . array ( [ <NUM_LIT> , <NUM_LIT> , <NUM_LIT> , <NUM_LIT> ] ) ) <EOL> def test_ic ( ) : <EOL> ics = [ aic , aicc , bic , hqic ] <EOL> ics_sig = [ aic_sigma , aicc_sigma , bic_sigma , hqic_sigma ] <EOL> for ic , ic_sig in zip ( ics , ics_sig ) : <EOL> assert_ ( ic ( np . array ( <NUM_LIT:2> ) , <NUM_LIT:10> , <NUM_LIT:2> ) . dtype == np . float , msg = repr ( ic ) ) <EOL> assert_ ( ic_sig ( np . array ( <NUM_LIT:2> ) , <NUM_LIT:10> , <NUM_LIT:2> ) . dtype == np . float , msg = repr ( ic_sig ) ) <EOL> assert_almost_equal ( ic ( - <NUM_LIT> / <NUM_LIT> * np . log ( <NUM_LIT> ) , <NUM_LIT:10> , <NUM_LIT:2> ) / <NUM_LIT:10> , <EOL> ic_sig ( <NUM_LIT:2> , <NUM_LIT:10> , <NUM_LIT:2> ) , <EOL> decimal = <NUM_LIT> ) <EOL> assert_almost_equal ( ic_sig ( np . log ( <NUM_LIT> ) , <NUM_LIT:10> , <NUM_LIT:2> , islog = True ) , <EOL> ic_sig ( <NUM_LIT:2> , <NUM_LIT:10> , <NUM_LIT:2> ) , <EOL> decimal = <NUM_LIT> ) <EOL> n , k = <NUM_LIT:10> , <NUM_LIT:2> <EOL> assert_almost_equal ( aic ( <NUM_LIT:0> , <NUM_LIT:10> , <NUM_LIT:2> ) , <NUM_LIT:2> * k , decimal = <NUM_LIT> ) <EOL> assert_almost_equal ( aicc ( <NUM_LIT:0> , <NUM_LIT:10> , <NUM_LIT:2> ) , <EOL> aic ( <NUM_LIT:0> , n , k ) + <NUM_LIT:2> * k * ( k + <NUM_LIT:1.> ) / ( n - k - <NUM_LIT:1.> ) , decimal = <NUM_LIT> ) <EOL> assert_almost_equal ( bic ( <NUM_LIT:0> , <NUM_LIT:10> , <NUM_LIT:2> ) , np . log ( n ) * k , decimal = <NUM_LIT> ) <EOL> assert_almost_equal ( hqic ( <NUM_LIT:0> , <NUM_LIT:10> , <NUM_LIT:2> ) , <NUM_LIT:2> * np . log ( np . log ( n ) ) * k , decimal = <NUM_LIT> ) <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> test_eval_measures ( ) <EOL> test_ic ( ) </s>
