#NEWLINE# def test_minimum_dimensions_batch_stream( self ) : #NEWLINE# #INDENT# stream = MinimumImageDimensions ( self . batch_stream , ( 4 , 5 ) , which_sources = ( 'source1' , ) ) #NEWLINE# it = stream . get_epoch_iterator ( ) #NEWLINE# for ( batch , shapes ) in zip ( it , partition_all ( self . batch_size , self . shapes ) ) : #NEWLINE# #INDENT# assert ( ( ( batch . shape [ 0 ] >= 4 ) and ( example . shape [ 1 ] >= 5 ) ) for example in batch [ 0 ] ) #NEWLINE# assert ( ( ( example . shape [ 1 ] == shp [ 0 ] ) and ( example . shape [ 0 ] == shp [ 1 ] ) ) for ( example , shp ) in zip ( batch [ 1 ] , shapes ) )