Turns out there's no way to test it from the public API. I'm going to replace this line by an assertion that `x_squared_norms is not None`.