mirror of
https://github.com/openswarm-ai/openswarm.git
synced 2026-08-20 03:35:40 +02:00
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
from backend.apps.agents.core.aux_llm import clean_short_label
|
|
|
|
|
|
def test_clean_label_passthrough():
|
|
assert clean_short_label("Travel planning") == "Travel planning"
|
|
|
|
|
|
def test_markdown_essay_becomes_short():
|
|
raw = "# Git Rebase Explained\nGit rebase is a way to integrate changes from one branch"
|
|
assert clean_short_label(raw) == "Git Rebase Explained"
|
|
|
|
|
|
def test_first_line_word_cap():
|
|
assert clean_short_label("one two three four five six") == "one two three four"
|
|
|
|
|
|
def test_char_cap_lands_on_word_boundary():
|
|
out = clean_short_label("supercalifragilistic expialidocious antidisestablishmentarianism", max_words=4)
|
|
assert len(out) <= 36
|
|
assert not out.endswith(" ")
|
|
|
|
|
|
def test_refusals_rejected():
|
|
for bad in [
|
|
"I cannot generate a name without more information",
|
|
"Sorry, there is no task to summarize",
|
|
"I'm unable to help with that",
|
|
"As an AI, I need more context",
|
|
"Unfortunately no information was provided",
|
|
]:
|
|
assert clean_short_label(bad) == ""
|
|
|
|
|
|
def test_quotes_and_bullets_stripped():
|
|
assert clean_short_label('"**Code review**"') == "Code review"
|
|
assert clean_short_label("- Sales dashboard.") == "Sales dashboard"
|
|
|
|
|
|
def test_empty_and_whitespace():
|
|
assert clean_short_label("") == ""
|
|
assert clean_short_label("\n\n \n") == ""
|
|
|
|
|
|
def test_ing_words_not_false_rejected():
|
|
assert clean_short_label("Investigating the bug") == "Investigating the bug"
|
|
assert clean_short_label("iOS app design") == "iOS app design"
|