import tensorflow as tf

    def testSeqAE(self):
        FLAGS.use_seq2seq_autoencoder = True
        graphs.VatxtModel().language_model_training()

    def testBidirLM(self):
        graphs.VatxtBidirModel().language_model_graph()

    def testBidirClassifier(self):
        at_methods = [None, 'rp', 'at', 'vat', 'atvat']
        for method in at_methods:
            FLAGS.adv_training_method = method
            with tf.Graph().as_default():
                graphs.VatxtBidirModel().classifier_graph()

                # Ensure variables have been reused
                # Embedding + 2 LSTM layers + hidden layers + logits layer
                expected_num_vars = 1 + 2 * 2 * FLAGS.rnn_num_layers + 2 * (
                    FLAGS.cl_num_layers) + 2
                self.assertEqual(len(tf.trainable_variables()), expected_num_vars)

    def testEvalGraph(self):
        _, _ = graphs.VatxtModel().eval_graph()
