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