#NEWLINE# def test_distrib_idx_in_full_out( self ) : #NEWLINE# #INDENT# size = 11 #NEWLINE# p = Problem ( root = Group ( ) , impl = impl ) #NEWLINE# top = p . root #NEWLINE# top . add ( 'C1' , InOutArrayComp ( size ) ) #NEWLINE# top . add ( 'C2' , DistribInputComp ( self ) ) #NEWLINE# top . connect ( 'C1.outvec' , 'C2.invec' ) #NEWLINE# p . setup ( check = False ) #NEWLINE# top . C1 . params [ 'invec' ] = np . array ( range ( size , 0 , ( - 1 ) ) , float ) #NEWLINE# p . run ( ) #NEWLINE# self . assertTrue ( all ( ( top . C2 . unknowns [ 'outvec' ] == ( np . array ( range ( size , 0 , ( - 1 ) ) , float ) * 4 ) ) ) )