Update test_pca.py