#NEWLINE# def test_make_swiss_roll( ) : #NEWLINE# #INDENT# ( X , t ) = make_swiss_roll ( n_samples = 5 , noise = 0.0 , random_state = 0 ) #NEWLINE# assert_equal ( X . shape , ( 5 , 3 ) , 'X shape mismatch' ) #NEWLINE# assert_equal ( t . shape , ( 5 , ) , 't shape mismatch' ) #NEWLINE# assert_array_almost_equal ( X [ : , 0 ] , ( t * np . cos ( t ) ) ) #NEWLINE# assert_array_almost_equal ( X [ : , 2 ] , ( t * np . sin ( t ) ) )