帮我写一个测试gpu渲染功能的python脚本

在Python中，你可以使用`pyOpenGL`库来测试GPU渲染功能。以下是一个简单的例子，这个脚本会创建一个窗口并尝试绘制一些基本图形：

from OpenGL.GLUT import *
from OpenGL.GL import *
from OpenGL.GLU import *

# 初始化GLUT
def init():
    glClearColor(0.2, 0.3, 0.4, 1.0)  # 设置背景颜色

# 绘制函数
def display():
    glClear(GL_COLOR_BUFFER_BIT)  # 清除屏幕
    glColor3f(1.0, 1.0, 1.0)  # 设置画笔颜色为白色
    glBegin(GL_TRIANGLES)  # 开始三角形绘制
    glVertex3f(-0.5, -0.5, 0.0)  # 顶点1
    glVertex3f(0.5, -0.5, 0.0)   # 顶点2
    glVertex3f(0.0, 0.5, 0.0)    # 顶点3
    glEnd()  # 结束绘制
    glutSwapBuffers()  # 交换双缓冲区完成渲染

if __name__ == '__main__':
    glutInit()
    glutInitDisplayMode(GLUT_DOUBLE | GLUT_RGB | GLUT_DEPTH)
    glutCreateWindow("GPU Rendering Test")

    glEnable(GL_DEPTH_TEST)  # 启用深度测试
    init()

    glutDisplayFunc(display)

    glutMainLoop()

运行此脚本时，如果GPU支持，它会在屏幕上显示一个白色的三角形。这表明GPU渲染功能正常工作。