<s> import numpy as np <EOL> from numpy . testing import assert_equal <EOL> from statsmodels . tools . catadd import add_indep <EOL> from scipy import linalg <EOL> def test_add_indep ( ) : <EOL> x1 = np . array ( [ <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:2> , <NUM_LIT:2> ] ) <EOL> x2 = np . array ( [ <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> , <NUM_LIT:1> ] ) <EOL> x0 = np . ones ( len ( x2 ) ) <EOL> x = np . column_stack ( [ x0 , x1 [ : , None ] * np . arange ( <NUM_LIT:3> ) , x2 [ : , None ] * np . arange ( <NUM_LIT:2> ) ] ) <EOL> varnames = [ '<STR_LIT>' ] + [ '<STR_LIT>' % i for i in np . arange ( <NUM_LIT:3> ) ] + [ '<STR_LIT>' % i for i in np . arange ( <NUM_LIT:2> ) ] <EOL> xo , vo = add_indep ( x , varnames ) <EOL> assert_equal ( xo , np . column_stack ( ( x0 , x1 , x2 ) ) ) <EOL> assert_equal ( ( linalg . svdvals ( x ) > <NUM_LIT> ) . sum ( ) , <NUM_LIT:3> ) <EOL> assert_equal ( vo , [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] ) <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> test_add_indep ( ) </s>
