Add principal eigenvector accuracy tests

[noahgolmant]

• Test accuracy of principal eigenvectors on some models
• Reformat code with black
• Fix stability of power iteration (facing NaNs in gradients sometimes)