Update light_decay_test.py