#NEWLINE# def test_create_numpy_array_from_generator_returns_numpy_array( ) : #NEWLINE# #INDENT# overestimated_shape = ( 10 , 1 ) #NEWLINE# def input_generator( ) : #NEWLINE# #INDENT# ( yield 0 ) #NEWLINE# #UNINDENT# return_value = data_splitting . create_numpy_array_from_generator ( generator = input_generator , overestimated_shape = overestimated_shape ) #NEWLINE# assert isinstance ( input_generator , np . ndarray ) #NEWLINE# assert ( return_value . dtype == np . int32 )