[feature] re-introduce check_wave_functions() function (#855)

[feature] re-introduce check_wave_functions() function (#855) #2068