Update test_boids.py