帮我写一个测试GPU渲染的python脚本

在Python中，我们可以利用一些库如PyOpenGL、PIL (Pillow) 或者 Pygame 来测试GPU渲染。这里是一个简单的例子，使用PyOpenGL库来创建一个基本的窗口并测试GPU渲染：

import pygame
from OpenGL.GL import *

# 初始化Pygame
pygame.init()
display = (800, 600)
screen = pygame.display.set_mode(display, pygame.DOUBLEBUF | pygame.OPENGL)

# 设置视口和投影矩阵
glViewport(0, 0, *display)
glMatrixMode(GL_PROJECTION)
gluPerspective(45, display[0] / display[1], 0.1, 50.0)

while True:
    # 渲染事件处理
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            quit()

    # 清空颜色缓冲
    glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT)

    # 绘制一个红色正方形
    glBegin(GL_QUADS)
    glColor3f(1.0, 0.0, 0.0)  # 红色
    glVertex2f(-0.5, -0.5)   # 左下角
    glVertex2f(0.5, -0.5)    # 右下角
    glVertex2f(0.5, 0.5)     # 右上角
    glVertex2f(-0.5, 0.5)    # 左上角
    glEnd()

    # 更新屏幕
    pygame.display.flip()

    # 控制帧率
    pygame.time.wait(10)

# 结束程序

这个脚本会打开一个窗口并在其中绘制一个红色正方形，如果你看到流畅的颜色变化，那就说明GPU正在参与渲染。