def test_pydotprint_cond_highlight():
 if (not theano.printing.pydot_imported):
  raise SkipTest('pydot not available')
 x = tensor.dvector()
 f = theano.function([x], (x * 2))
 f([1, 2, 3, 4])
 s = StringIO()
 new_handler = logging.StreamHandler(s)
 new_handler.setLevel(logging.DEBUG)
 orig_handler = theano.logging_default_handler
 theano.theano_logger.removeHandler(orig_handler)
 theano.theano_logger.addHandler(new_handler)
 try:
  theano.printing.pydotprint(f, cond_highlight=True, print_output_file=False)
 finally:
  theano.theano_logger.addHandler(orig_handler)
  theano.theano_logger.removeHandler(new_handler)
 assert (s.getvalue() == 'pydotprint: cond_highlight is set but there is no IfElse node in the graph\n')

def test_pydotprint_cond_highlight ( ) : if ( not theano . printing . pydot_imported ) : raise SkipTest ( 'pydot not available' ) x = tensor . dvector ( ) f = theano . function ( [ x ] , ( x * 2 ) ) f ( [ 1 , 2 , 3 , 4 ] ) s = StringIO ( ) new_handler = logging . StreamHandler ( s ) new_handler . setLevel ( logging . DEBUG ) orig_handler = theano . logging_default_handler theano . theano_logger . removeHandler ( orig_handler ) theano . theano_logger . addHandler ( new_handler ) try : theano . printing . pydotprint ( f , cond_highlight = True , print_output_file = False ) finally : theano . theano_logger . addHandler ( orig_handler ) theano . theano_logger . removeHandler ( new_handler ) assert ( s . getvalue ( ) == 'pydotprint: cond_highlight is set but there is no IfElse node in the graph\n' )
this is a really partial test .