写一个交通灯的测试代码

这里给出一个交通灯的简单测试代码，使用 Python 语言编写。

# 定义交通灯状态
class TrafficLight:
    def __init__(self):
        self.color = 'red'
        
    def change_color(self):
        if self.color == 'red':
            self.color = 'green'
        elif self.color == 'green':
            self.color = 'yellow'
        elif self.color == 'yellow':
            self.color = 'red'

# 测试交通灯状态变化
tl = TrafficLight()

# 初始状态应该是红灯
assert tl.color == 'red', "Initial color should be red"

# 先变成绿灯，再变成黄灯，最后变成红灯
tl.change_color()
assert tl.color == 'green', "Color should be green after first change"
tl.change_color()
assert tl.color == 'yellow', "Color should be yellow after second change"
tl.change_color()
assert tl.color == 'red', "Color should be red after third change"

print("All tests passed!")

该测试代码定义了一个 TrafficLight 类，该类具有一个 color 属性表示当前交通灯的颜色。在实例化交通灯对象时，初始状态设置为红灯。

测试代码通过执行交通灯状态变化方法 change_color()，模拟交通灯的状态变化，并对变化后的状态进行断言测试。一共执行了三次状态变化，分别从红灯变成绿灯、从绿灯变成黄灯、从黄灯变成红灯。

如果所有的断言测试都通过，那么输出 "All tests passed!"。