def METHOD_NAME() -> None:
    """Check that scripting actually occurs"""
    @compile_mode("script")
    class ScriptSubmod(torch.nn.Module):
        def forward(self, x):
            if x.shape[0] == 7:
                return x.new_zeros(8)
            else:
                return x
    class ParentMod(torch.nn.Module):
        def __init__(self) -> None:
            super().__init__()
            self.child = ScriptSubmod()
        def forward(self, x):
            return self.child(x)
    parent = ParentMod()
    assert parent(torch.randn(7, 4)).shape == (8,)
    parent_trace = trace_module(parent, inputs={"forward": (torch.randn(7, 4),)})  # get the conditional behaviour
    # Does it get the behaviour it was traced for?
    assert parent_trace(torch.randn(7, 4)).shape == (8,)
    # Does it get the conditional that should have been scripted?
    x = torch.randn(5, 7)
    assert torch.allclose(parent(x), x)
    assert torch.allclose(parent_trace(x), x)