import tensorflow as tf


        # 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)

    def testLanguageModelGraph(self):
        train_op, _, _ = graphs.VatxtModel().language_model_training()
        with self.test_session() as sess:
            sess.run(tf.global_variables_initializer())
            tf.train.start_queue_runners(sess)
            sess.run(train_op)

    def testMulticlass(self):
        FLAGS.num_classes = 10
        graphs.VatxtModel().classifier_graph()
