Spaces:
Sleeping
Sleeping
File size: 1,709 Bytes
76684fa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
import unittest
import os
import sys
# 添加 src 目录到模块搜索路径，以便可以导入 src 目录中的模块
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../src')))
from layout_manager import LayoutManager
from data_structures import SlideContent
class TestLayoutManager(unittest.TestCase):
"""
测试 LayoutManager 类，验证布局分配逻辑是否正确。
"""
def setUp(self):
# 模拟布局映射字典
layout_mapping = {
"Title 1": 1,
"Title, Content 0": 2,
"Title, Content, Picture 2": 8
}
self.layout_manager = LayoutManager(layout_mapping)
def test_assign_layout_title_only(self):
content = SlideContent(title="Only Title")
layout_id, layout_name = self.layout_manager.assign_layout(content)
self.assertEqual(layout_id, 1)
self.assertEqual(layout_name, "Title 1")
def test_assign_layout_title_and_content(self):
content = SlideContent(title="Title with Content", bullet_points=[{'text': "Content Bullet", 'level': 0}])
layout_id, layout_name = self.layout_manager.assign_layout(content)
self.assertEqual(layout_id, 2)
self.assertEqual(layout_name, "Title, Content 0")
def test_assign_layout_title_content_and_image(self):
content = SlideContent(title="Full Slide", bullet_points=[{'text': "Full Content", 'level': 0}], image_path="images/test.png")
layout_id, layout_name = self.layout_manager.assign_layout(content)
self.assertEqual(layout_id, 8)
self.assertEqual(layout_name, "Title, Content, Picture 2")
if __name__ == "__main__":
unittest.main()
|