def METHOD_NAME(self):
    x = torch.rand(2, 2, 2)
    model = OpModel(torch.nn.functional.adaptive_avg_pool2d, [2, 2]).eval()
    nodes = get_model_onnx_nodes(model, x)
    assert nodes[-1].op_type == 'AveragePool'