mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
96 lines
3.1 KiB
Python
96 lines
3.1 KiB
Python
"""Tests for edge cases in observation parsing."""
|
|
|
|
import pytest
|
|
|
|
from basic_memory.markdown import ParseError, KnowledgeParser
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_observation_empty_input():
|
|
"""Test handling of empty input."""
|
|
parser = KnowledgeParser()
|
|
assert await parser.parse_observation("") is None
|
|
assert await parser.parse_observation(" ") is None
|
|
assert await parser.parse_observation("\n") is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_observation_invalid_context():
|
|
"""Test handling of invalid context format."""
|
|
parser = KnowledgeParser()
|
|
obs = await parser.parse_observation("- [test] Content (unclosed")
|
|
assert obs is not None
|
|
assert obs.content == "Content (unclosed"
|
|
assert obs.context is None
|
|
|
|
obs = await parser.parse_observation("- [test] Content (with) extra) parens)")
|
|
assert obs is not None
|
|
assert obs.content == "Content"
|
|
assert obs.context == "with) extra) parens"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_observation_complex_format():
|
|
"""Test parsing complex observation formats."""
|
|
# Test multiple nested tags and spaces
|
|
parser = KnowledgeParser()
|
|
obs = await parser.parse_observation("- [complex test] This is #tag1#tag2 with #tag3 content")
|
|
assert obs is not None
|
|
assert obs.category == "complex test"
|
|
assert set(obs.tags) == {"tag1", "tag2", "tag3"} # pyright: ignore [reportArgumentType]
|
|
assert obs.content == "This is with content"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_observation_exception_handling():
|
|
"""Test general error handling in observation parsing."""
|
|
# Test with a problematic regex pattern that could cause catastrophic backtracking
|
|
long_input = "[test] " + "a" * 1000000 # Very long input
|
|
parser = KnowledgeParser()
|
|
|
|
assert await parser.parse_observation(long_input) is None
|
|
|
|
# Test with invalid types
|
|
assert await parser.parse_observation(None) is None # type: ignore
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_observation_malformed_category():
|
|
"""Test handling of malformed category brackets."""
|
|
parser = KnowledgeParser()
|
|
with pytest.raises(ParseError, match="unclosed category"):
|
|
await parser.parse_observation("- [test Content")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_observation_empty_category():
|
|
parser = KnowledgeParser()
|
|
obs = await parser.parse_observation("- [] Empty category")
|
|
|
|
assert obs is not None
|
|
assert obs.category is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_observation_whitespace():
|
|
"""Test handling of whitespace."""
|
|
# Valid whitespace cases
|
|
parser = KnowledgeParser()
|
|
obs = await parser.parse_observation("- [test] Content")
|
|
assert obs is not None
|
|
assert obs.content == "Content"
|
|
|
|
# Test individual whitespace chars
|
|
test_chars = {
|
|
" ": "space",
|
|
"\t": "tab",
|
|
#'\n': 'newline', # newline should be end of content
|
|
"\r": "return",
|
|
}
|
|
|
|
for char, name in test_chars.items():
|
|
content = f"- [test] Content{char}with{char}{name}"
|
|
obs = await parser.parse_observation(content)
|
|
assert obs is not None
|
|
assert obs.content == f"Content with {name}"
|