feat: 修复代码审核报告问题
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
"""分类/标签模块测试"""
|
||||
from app.tagger import classify_article, tag_article, tag_articles
|
||||
from models import EnrichedArticle, Taxonomy
|
||||
|
||||
|
||||
def test_classify_article(db):
|
||||
categories = [
|
||||
Taxonomy(name="科技", kind="category", keywords=["AI", "大模型", "芯片"]),
|
||||
Taxonomy(name="财经", kind="category", keywords=["股市", "基金", "财报"]),
|
||||
]
|
||||
article = EnrichedArticle(
|
||||
rk_article_id=1,
|
||||
title="OpenAI 发布新一代大模型",
|
||||
)
|
||||
assert classify_article(article, categories) == "科技"
|
||||
|
||||
|
||||
def test_tag_article(db):
|
||||
tags = [
|
||||
Taxonomy(name="人工智能", kind="tag", keywords=["AI", "人工智能", "大模型"]),
|
||||
Taxonomy(name="半导体", kind="tag", keywords=["芯片", "半导体"]),
|
||||
]
|
||||
article = EnrichedArticle(
|
||||
rk_article_id=1,
|
||||
title="OpenAI 发布新一代大模型",
|
||||
)
|
||||
result = tag_article(article, tags)
|
||||
assert "人工智能" in result
|
||||
|
||||
|
||||
def test_tag_articles_integration(db):
|
||||
db.add_all([
|
||||
Taxonomy(name="科技", kind="category", keywords=["AI", "大模型"]),
|
||||
Taxonomy(name="人工智能", kind="tag", keywords=["AI", "大模型"]),
|
||||
])
|
||||
article = EnrichedArticle(rk_article_id=1, title="OpenAI 发布 GPT-5 大模型")
|
||||
db.add(article)
|
||||
db.commit()
|
||||
|
||||
count = tag_articles(db)
|
||||
assert count == 1
|
||||
assert article.category == "科技"
|
||||
assert "人工智能" in article.tags
|
||||
Reference in New Issue
Block a user