def METHOD_NAME(self):
    valid, test = testing_utils.train_model(
        Opt(
            dict(
                task='integration_tests:classifier',
                model='projects.style_gen.classifier:ClassifierAgent',
                classes=['one', 'zero'],
                optimizer='adamax',
                truncate=8,
                learningrate=7e-3,
                batchsize=16,
                num_epochs=5,
                n_layers=1,
                n_heads=1,
                ffn_size=32,
                embedding_size=32,
            )
        )
    )
    self.assertEqual(valid['accuracy'], 1.0)
    self.assertEqual(test['accuracy'], 1.0)