写一个关于小球运动的子程序

编写一个关于小球运动的子程序，通常涉及到物理模拟或游戏开发中的基本元素。这个子程序可能会处理以下几个步骤：

class Ball:
    def __init__(self, x_position, y_position, speed_x, speed_y):
        self.x_position = x_position
        self.y_position = y_position
        self.speed_x = speed_x
        self.speed_y = speed_y

    def move(self, time_step):
        # 根据加速度计算新位置
        self.x_position += self.speed_x * time_step
        self.y_position += self.speed_y * time_step
        
        # 检查边界条件并更新速度方向（例如碰撞检测）
        if self.x_position < 0 or self.x_position > screen_width:
            self.speed_x *= -1  # 如果碰到边缘反弹
        if self.y_position < 0:
            self.speed_y *= -1  # 同理，如果碰到顶部
            
    def display(self, canvas):
        # 绘制小球到屏幕上的指定位置
        draw_circle(canvas, self.x_position, self.y_position, ball_radius)

# 使用示例
ball = Ball(0, screen_height / 2, 5, 0)  # 初始化小球位置和初始速度
while True:
    ball.move(time_step)  # 更新小球位置
    ball.display(screen)  # 显示小球