import tensorflow as tf

    def tearDownClass(cls):
        shutil.rmtree(FLAGS.data_dir)

    def setUp(self):
        # Reset FLAGS
        FLAGS.rnn_num_layers = 1
        FLAGS.sync_replicas = False
        FLAGS.adv_training_method = None
        FLAGS.num_candidate_samples = -1
        FLAGS.num_classes = 2
        FLAGS.use_seq2seq_autoencoder = False

        # Reset Graph
        tf.reset_default_graph()

    def testClassifierGraph(self):
        FLAGS.rnn_num_layers = 2
        model = graphs.VatxtModel()
        train_op, _, _ = model.classifier_training()
        # Pretrained vars: embedding + LSTM layers
        self.assertEqual(
            len(model.pretrained_variables), 1 + 2 * FLAGS.rnn_num_layers)
        with self.test_session() as sess:
            sess.run(tf.global_variables_initializer())
            tf.train.start_queue_runners(sess)
            sess.run(train_op)
