def METHOD_NAME(self) -> None:
    x = torch.randn((2, 3, 4, 5))
    actual = rearrange(x, 'b h w c -> b 1 h w 1 c')
    expected = x.unsqueeze(1).unsqueeze(-2)
    torch.testing.assert_close(actual, expected)