def METHOD_NAME() -> None:
    theta_3d = create_theta_3d()
    N, C, D, W, H = len(theta_3d), 3, 4, 5, 6
    data_size = (D, W, H)
    for align_corners in (0, 1):
        node = onnx.helper.make_node(
            "AffineGrid",
            inputs=["theta", "size"],
            outputs=["grid"],
            align_corners=align_corners,
        )
        original_grid = construct_original_grid(data_size, align_corners)
        grid = apply_affine_transform(theta_3d, original_grid)
        test_name = "test_affine_grid_3d"
        if align_corners == 1:
            test_name += "_align_corners"
        expect(
            node,
            inputs=[theta_3d, np.array([N, C, D, W, H], dtype=np.int64)],
            outputs=[grid],
            name=test_name,
        )