1秒后如何更新pygame屏幕

Question

我正在尝试创建一个执行以下操作的pygame程序

1. 创建一个白色窗口，等待1秒钟。
2. 创建一个红色正方形，等待1秒钟。
3. 创建一个白色窗口，等待1秒钟。
4. 退出程序

当我运行以下程序时，该窗口仅打开3秒钟，并显示一个空白窗口并结束它。

我该如何解决这个问题？

import pygame, sys, time
from pygame.locals import *

pygame.init()

screen = pygame.display.set_mode((1000, 500))
pygame.display.set_caption("testgame2")

white = (255, 255, 255)
red = (255, 0, 0)

screen.fill(white)
pygame.display.update()
time.sleep(1)

pygame.draw.rect(screen, red, (0, 0, 250, 250))
pygame.display.update()
time.sleep(1)

screen.fill(white)
pygame.display.update()
time.sleep(1)

pygame.quit()
sys.exit()