Add a test for week_start and week_end