mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
Paul Hernandez
5b4f0eafcc
* configure logging * set mcp output logging also * fix type check errors * fix type check * rename Permalink schema type * fix type errors * add typechecks to ci workflow * pytest coverage setup * add tests for status cli * sync tests coverage * watch_service test coverage * tests for tool_utils.py * clean up imports * file_utils coverage * markdown plugins coverage * 99% test coverage * more test coverage, remove ObservationCategory * more tool coverage * fix type-check * format, upgrade deps --------- Co-authored-by: phernandez <phernandez@basicmachines.co>
127 lines
3.3 KiB
Python
127 lines
3.3 KiB
Python
"""Tests for search schemas."""
|
|
|
|
from datetime import datetime
|
|
|
|
from basic_memory.schemas.search import (
|
|
SearchItemType,
|
|
SearchQuery,
|
|
SearchResult,
|
|
SearchResponse,
|
|
AdvancedSearchQuery,
|
|
)
|
|
|
|
|
|
def test_search_modes():
|
|
"""Test different search modes."""
|
|
# Exact permalink
|
|
query = SearchQuery(permalink="specs/search")
|
|
assert query.permalink == "specs/search"
|
|
assert query.text is None
|
|
|
|
# Pattern match
|
|
query = SearchQuery(permalink="specs/*")
|
|
assert query.permalink == "specs/*"
|
|
assert query.text is None
|
|
|
|
# Text search
|
|
query = SearchQuery(text="search implementation")
|
|
assert query.text == "search implementation"
|
|
assert query.permalink is None
|
|
|
|
|
|
def test_search_filters():
|
|
"""Test search result filtering."""
|
|
query = SearchQuery(
|
|
text="search",
|
|
types=[SearchItemType.ENTITY],
|
|
entity_types=["component"],
|
|
after_date=datetime(2024, 1, 1),
|
|
)
|
|
assert query.types == [SearchItemType.ENTITY]
|
|
assert query.entity_types == ["component"]
|
|
assert query.after_date == "2024-01-01T00:00:00"
|
|
|
|
|
|
def test_search_result():
|
|
"""Test search result structure."""
|
|
result = SearchResult(
|
|
id=1,
|
|
type=SearchItemType.ENTITY,
|
|
score=0.8,
|
|
metadata={"entity_type": "component"},
|
|
permalink="specs/search",
|
|
file_path="specs/search.md",
|
|
)
|
|
assert result.id == 1
|
|
assert result.type == SearchItemType.ENTITY
|
|
assert result.score == 0.8
|
|
assert result.metadata == {"entity_type": "component"}
|
|
|
|
|
|
def test_observation_result():
|
|
"""Test observation result fields."""
|
|
result = SearchResult(
|
|
id=1,
|
|
permalink="specs/search",
|
|
file_path="specs/search.md",
|
|
type=SearchItemType.OBSERVATION,
|
|
score=0.5,
|
|
metadata={},
|
|
entity_id=123,
|
|
category="tech",
|
|
)
|
|
assert result.entity_id == 123
|
|
assert result.category == "tech"
|
|
|
|
|
|
def test_relation_result():
|
|
"""Test relation result fields."""
|
|
result = SearchResult(
|
|
id=1,
|
|
permalink="specs/search",
|
|
file_path="specs/search.md",
|
|
type=SearchItemType.RELATION,
|
|
score=0.5,
|
|
metadata={},
|
|
from_id=123,
|
|
to_id=456,
|
|
relation_type="depends_on",
|
|
)
|
|
assert result.from_id == 123
|
|
assert result.to_id == 456
|
|
assert result.relation_type == "depends_on"
|
|
|
|
|
|
def test_search_response():
|
|
"""Test search response wrapper."""
|
|
results = [
|
|
SearchResult(
|
|
id=1,
|
|
permalink="specs/search",
|
|
file_path="specs/search.md",
|
|
type=SearchItemType.ENTITY,
|
|
score=0.8,
|
|
metadata={},
|
|
),
|
|
SearchResult(
|
|
id=2,
|
|
permalink="specs/search",
|
|
file_path="specs/search.md",
|
|
type=SearchItemType.ENTITY,
|
|
score=0.6,
|
|
metadata={},
|
|
),
|
|
]
|
|
response = SearchResponse(results=results)
|
|
assert len(response.results) == 2
|
|
assert response.results[0].score > response.results[1].score
|
|
|
|
|
|
def test_advanced_search():
|
|
"""Test advanced search query."""
|
|
query = AdvancedSearchQuery(
|
|
query="title:search AND content:implementation", types=[SearchItemType.ENTITY]
|
|
)
|
|
assert query.query == "title:search AND content:implementation"
|
|
assert query.types == [SearchItemType.ENTITY]
|