Update test_noise_estimate.py