Update test_router.py

- Refine something