Yong-yuan-X
e74b0e1e93
Add a pytest-discovered tests/conftest.py for the last30days scripts path and remove duplicate per-file sys.path.insert boilerplate from tests. Normalize affected imports to rely on the shared scripts path and remove the now-unneeded E402 suppressions.
40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
import unittest
|
|
|
|
from lib import query
|
|
|
|
|
|
class QueryV3Tests(unittest.TestCase):
|
|
def test_extract_core_subject_strips_prefix_and_noise(self):
|
|
result = query.extract_core_subject("What are the best Claude Code skills for startups?")
|
|
self.assertEqual("claude code startups", result)
|
|
|
|
def test_extract_core_subject_supports_suffix_stripping_and_max_words(self):
|
|
result = query.extract_core_subject(
|
|
"How to use Claude Code prompting techniques",
|
|
strip_suffixes=True,
|
|
max_words=2,
|
|
)
|
|
self.assertEqual("claude code", result)
|
|
|
|
def test_extract_core_subject_preserves_original_when_noise_removes_everything(self):
|
|
result = query.extract_core_subject("all tokens", noise=frozenset({"all", "tokens"}))
|
|
self.assertEqual("all tokens", result)
|
|
|
|
def test_extract_core_subject_handles_empty_query_and_custom_noise(self):
|
|
self.assertEqual("", query.extract_core_subject(" "))
|
|
result = query.extract_core_subject(
|
|
"OpenClaw release notes",
|
|
noise=frozenset({"release"}),
|
|
max_words=2,
|
|
)
|
|
self.assertEqual("openclaw notes", result)
|
|
|
|
def test_extract_compound_terms_finds_hyphenated_and_title_cased_phrases(self):
|
|
terms = query.extract_compound_terms("Best multi-agent patterns in Claude Code and React Native")
|
|
self.assertIn("multi-agent", terms)
|
|
self.assertIn("Claude Code", terms)
|
|
self.assertIn("React Native", terms)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|