Update test_ray_tracing.py