content-find-agent-new/tests/test_search_previous_step.py

"""标签 labeling 修复:游走标签搜索结果标 hashtag_to_query,首轮标 search_query_direct。"""

from content_agent.integrations.crawapi_http import search_previous_discovery_step


def test_tag_query_labeled_as_walk():
    assert search_previous_discovery_step({"search_query_generation_method": "tag_query"}) == "hashtag_to_query"


def test_first_round_methods_labeled_search_query_direct():
    for method in ["seed_term", "item_single", "piaoquan_topic_point", "query_seed_point", "category_leaf_element", None]:
        assert search_previous_discovery_step({"search_query_generation_method": method}) == "search_query_direct"
    assert search_previous_discovery_step({}) == "search_query_direct"