mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
"""Tests for the markdown entity parser."""
|
|
|
|
import pytest
|
|
|
|
from basic_memory.markdown import Observation
|
|
from basic_memory.markdown.parser import EntityParser, ParseError
|
|
|
|
|
|
def test_parse_observation_basic():
|
|
"""Test basic observation parsing with category and tags."""
|
|
parser = EntityParser()
|
|
|
|
obs = Observation.from_line("- [design] Core feature #important #mvp")
|
|
|
|
assert obs is not None
|
|
assert obs.category == "design"
|
|
assert obs.content == "Core feature"
|
|
assert set(obs.tags) == {"important", "mvp"}
|
|
assert obs.context is None
|
|
|
|
|
|
def test_parse_observation_with_context():
|
|
"""Test observation parsing with context in parentheses."""
|
|
parser = EntityParser()
|
|
|
|
obs = Observation.from_line(
|
|
"- [feature] Authentication system #security #auth (Required for MVP)"
|
|
)
|
|
assert obs is not None
|
|
assert obs.category == "feature"
|
|
assert obs.content == "Authentication system"
|
|
assert set(obs.tags) == {"security", "auth"}
|
|
assert obs.context == "Required for MVP"
|
|
|
|
|
|
def test_parse_observation_edge_cases():
|
|
"""Test observation parsing edge cases."""
|
|
parser = EntityParser()
|
|
|
|
# Multiple word tags
|
|
obs = Observation.from_line("- [tech] Database #high-priority #needs-review")
|
|
|
|
assert obs is not None
|
|
|
|
assert set(obs.tags) == {"high-priority", "needs-review"}
|
|
|
|
# Multiple word category
|
|
obs = Observation.from_line("- [user experience] Design #ux")
|
|
assert obs is not None
|
|
assert obs.category == "user experience"
|
|
|
|
# Parentheses in content shouldn't be treated as context
|
|
obs = Observation.from_line("- [code] Function (x) returns y #function")
|
|
assert obs is not None
|
|
assert obs.content == "Function (x) returns y"
|
|
assert obs.context is None
|
|
|
|
# Multiple hashtags together
|
|
obs = Observation.from_line("- [test] Feature #important#urgent#now")
|
|
assert obs is not None
|
|
assert set(obs.tags) == {"important", "urgent", "now"}
|
|
|
|
|
|
def test_parse_observation_errors():
|
|
"""Test error handling in observation parsing."""
|
|
parser = EntityParser()
|
|
|
|
# Missing category brackets
|
|
with pytest.raises(ParseError, match="missing category"):
|
|
Observation.from_line("- Design without brackets #test")
|
|
|
|
# Unclosed category
|
|
with pytest.raises(ParseError, match="unclosed category"):
|
|
Observation.from_line("- [design Core feature #test")
|