Function test_task_101

data/unit_test/tests.py:1162–1188

(trajectory)

def test_task_101(trajectory):
    image_path = "./output/101.png"
    ref_path = "./data/101.jpeg"
    
    # 打开目标图像
    image = Image.open(image_path)
    
    # 将图像转换为灰度图
    gray_image = image.convert('L')
    
    # 将灰度图转换为numpy数组
    gray_array = np.array(gray_image)
    
    # 定义黑色像素的阈值（例如，灰度值小于等于50的像素被认为是黑色）
    black_threshold = 50
    
    # 计算黑色像素的数量
    black_pixels = np.sum(gray_array <= black_threshold)
    
    # 计算总像素数量
    total_pixels = gray_array.size
    
    # 计算黑色像素比例
    black_ratio = black_pixels / total_pixels
    print(black_ratio)
    # 检查图像中黑色像素的比例是否超过30%
    assert black_ratio > 0.5, "The black pixel percentage does not exceed 50%."

def test_task_102(trajectory):
    image_path = "./output/102.png"