so sqrt(square(-i)) = +-i, one of which is x
So no, it’s not unconditionally correct either.
But consider sqrt(i) = sqrt(exp(i\pi/2)). That's exp(i\pi/4). Your rule would give 1 as the answer. It's not helpful for a serious math system to give that answer to this problem.
When I square 1 I don't get i.