def test_main1():
 import tempfile
 tempdir = tempfile.mkdtemp()
 arglist = ('--data-dir %s -e 2 --iter-num=1 -v -v --tensorboard-verbose=1 train 5000' % tempdir)
 arglist = arglist.split(' ')
 tf.reset_default_graph()
 ts2s = CommandLine(arglist=arglist)
 assert os.path.exists(ts2s.weights_output_fn)
 os.system(('rm -rf %s' % tempdir))

def test_main1 ( ) : import tempfile tempdir = tempfile . mkdtemp ( ) arglist = ( '--data-dir %s -e 2 --iter-num=1 -v -v --tensorboard-verbose=1 train 5000' % tempdir ) arglist = arglist . split ( ' ' ) tf . reset_default_graph ( ) ts2s = CommandLine ( arglist = arglist ) assert os . path . exists ( ts2s . weights_output_fn ) os . system ( ( 'rm -rf %s' % tempdir ) )
integration test - training .