def METHOD_NAME(self, test_data, device, args):
    # test data
    metrics = self._test(test_data, device, args)
    test_acc, test_num_sample, test_loss = (
        metrics["test_acc"],
        metrics["test_total"],
        metrics["test_loss"],
    )
    # test on test dataset
    if self.args.enable_wandb:
        wandb.log({"Test/Acc": test_acc, "round": args.round_idx})
        wandb.log({"Test/Loss": test_loss, "round": args.round_idx})
    mlops.log({"Test/Acc": test_acc, "round": args.round_idx})
    mlops.log({"Test/Loss": test_loss, "round": args.round_idx})
    stats = {"test_acc": test_acc, "test_loss": test_loss}
    logging.info(stats)