Update unittest for compute_pce_matrix (better coverage)