mirror of
https://github.com/CyC2018/Interview-Notebook.git
synced 2026-11-06 07:34:10 +00:00
32 lines
1.5 KiB
Python
32 lines
1.5 KiB
Python
"""Regression checks for generated search anchors and asset collection."""
|
|
import sys
|
|
import unittest
|
|
from collections import Counter
|
|
from pathlib import Path
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parent))
|
|
from site_assets import heading_slug, referenced_paths, search_document
|
|
|
|
|
|
class SearchIndexTests(unittest.TestCase):
|
|
def test_docsify_numeric_duplicate_and_chinese_punctuation(self):
|
|
counts = Counter()
|
|
self.assertEqual(heading_slug('1. 慢开始与拥塞避免', counts), '_1-慢开始与拥塞避免')
|
|
self.assertEqual(heading_slug('三、检索增强生成', counts), '三、检索增强生成')
|
|
self.assertEqual(heading_slug('TCP 拥塞控制', counts), 'tcp-拥塞控制')
|
|
self.assertEqual(heading_slug('TCP 拥塞控制', counts), 'tcp-拥塞控制-1')
|
|
|
|
def test_headings_in_code_do_not_change_real_anchors(self):
|
|
doc = search_document(Path('示例.md'), '# 示例\n\n```python\n# TCP 拥塞控制\n```\n\n## TCP 拥塞控制\n慢开始\n')
|
|
sections = [s for s in doc['sections'] if s['id'] == 'tcp-拥塞控制']
|
|
self.assertEqual(len(sections), 1)
|
|
self.assertEqual(sections[0]['text'], '慢开始')
|
|
|
|
def test_comments_and_example_asset_urls_are_not_published(self):
|
|
content = '<!-- <img src="/secret.png"> -->\n```html\n<img src="/example.png">\n```\n<img src="/notes/pics/actual.png">\n[正文](/notes/示例.md)\n'
|
|
self.assertEqual(set(referenced_paths(content)), {'notes/pics/actual.png', 'notes/示例.md'})
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|