2569718930@qq.com
40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
from scripts.sync_forum_topics_mtproto import build_city_thread_mapping
|
|
|
|
|
|
def test_build_city_thread_mapping_normalizes_hashtag_topic_names():
|
|
topics = [
|
|
{"id": 101, "title": "#seoul"},
|
|
{"id": 102, "title": "#HongKong"},
|
|
{"id": 103, "title": "#TelAviv"},
|
|
{"id": 104, "title": "#NewYork"},
|
|
{"id": 105, "title": "#SanFrancisco"},
|
|
{"id": 106, "title": "#LauFauShan"},
|
|
]
|
|
|
|
mapping = build_city_thread_mapping(topics)
|
|
|
|
assert mapping == {
|
|
"seoul": 101,
|
|
"hong kong": 102,
|
|
"tel aviv": 103,
|
|
"new york": 104,
|
|
"san francisco": 105,
|
|
"shenzhen": 106,
|
|
}
|
|
|
|
|
|
def test_build_city_thread_mapping_supports_flagged_topic_titles():
|
|
topics = [
|
|
{"id": 201, "title": "🇰🇷 Seoul"},
|
|
{"id": 202, "title": "🇨🇳 Shenzhen / Lau Fau Shan"},
|
|
{"id": 203, "title": "🇺🇸 Los Angeles"},
|
|
]
|
|
|
|
mapping = build_city_thread_mapping(topics)
|
|
|
|
assert mapping == {
|
|
"seoul": 201,
|
|
"shenzhen": 202,
|
|
"los angeles": 203,
|
|
}
|