84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
"""
|
||
inkycal_today unittest
|
||
"""
|
||
import logging
|
||
import unittest
|
||
|
||
from inkycal.modules import Today
|
||
from inkycal.modules.inky_image import Inkyimage, image_to_palette
|
||
from tests import Config
|
||
|
||
merge = Inkyimage.merge
|
||
|
||
sample_url = Config.SAMPLE_ICAL_URL
|
||
|
||
logger = logging.getLogger(__name__)
|
||
logging.basicConfig(level=logging.DEBUG)
|
||
|
||
# /workspaces/Inkycal/venv/bin/python3 -m pytest tests/test_inkycal_today.py::TestToday::test_generate_image -v -s
|
||
tests = [
|
||
{
|
||
"name": "Today",
|
||
"config": {
|
||
"size": [528, 343],
|
||
"week_starts_on": "Monday",
|
||
"show_events": True,
|
||
"ical_urls": sample_url,
|
||
"ical_files": None,
|
||
"date_format": "D MMM", "time_format": "HH:mm",
|
||
"padding_x": 10, "padding_y": 10, "fontsize": 14, "language": "zh",
|
||
"font": "NotoSansCJKsc-Regular",
|
||
}
|
||
},
|
||
]
|
||
|
||
|
||
class TestToday(unittest.TestCase):
|
||
|
||
def test_generate_image(self):
|
||
output_dir = Config.OUTPUT_DIR
|
||
import os
|
||
from PIL import Image, ImageDraw
|
||
import numpy as np
|
||
|
||
os.makedirs(output_dir, exist_ok=True)
|
||
test_num = 0
|
||
for test in tests:
|
||
print(f'test {tests.index(test) + 1} generating image..', end="")
|
||
module = Today(test)
|
||
im_black, im_colour = module.generate_image()
|
||
print('OK')
|
||
|
||
# 创建最终的三色图像
|
||
# 在 E-Paper 上: im_black 的黑色像素显示黑色, im_colour 的黑色像素显示红色
|
||
result = Image.new('RGB', im_black.size, 'white')
|
||
result_array = np.array(result)
|
||
black_array = np.array(im_black)
|
||
colour_array = np.array(im_colour)
|
||
|
||
# 使用阈值处理抗锯齿：灰度值 < 128 的视为"黑色"
|
||
# im_black 上的深色区域 -> 黑色 (0, 0, 0)
|
||
black_gray = black_array[:,:,0] # 取灰度值（RGB相同）
|
||
black_mask = black_gray < 128
|
||
result_array[black_mask] = [0, 0, 0]
|
||
|
||
# im_colour 上的深色区域 -> 红色 (255, 0, 0)
|
||
colour_gray = colour_array[:,:,0]
|
||
colour_mask = colour_gray < 128
|
||
result_array[colour_mask] = [255, 0, 0]
|
||
|
||
# 保存最终图像
|
||
final_image = Image.fromarray(result_array)
|
||
output_path = os.path.join(output_dir, f"today_test_{test_num}.png")
|
||
final_image.save(output_path)
|
||
print(f' -> Saved to {output_path}')
|
||
|
||
# 统计颜色
|
||
red_count = np.sum(colour_mask)
|
||
black_count = np.sum(black_mask)
|
||
print(f' 🔴 Red pixels: {red_count}, ⚫ Black pixels: {black_count}')
|
||
|
||
test_num += 1
|
||
|
||
if Config.USE_PREVIEW:
|
||
final_image.show() |