mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
bm-claudeai
020957cd76
Signed-off-by: phernandez <paul@basicmachines.co> Co-authored-by: phernandez <paul@basicmachines.co> Co-authored-by: Claude <noreply@anthropic.com>
58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
"""Tests for importer utility functions."""
|
|
|
|
from datetime import datetime
|
|
|
|
from basic_memory.importers.utils import clean_filename, format_timestamp
|
|
|
|
|
|
def test_clean_filename():
|
|
"""Test clean_filename utility function."""
|
|
# Test with normal string
|
|
assert clean_filename("Hello World") == "Hello_World"
|
|
|
|
# Test with punctuation
|
|
assert clean_filename("Hello, World!") == "Hello_World"
|
|
|
|
# Test with special characters
|
|
assert clean_filename("File[1]/with\\special:chars") == "File_1_with_special_chars"
|
|
|
|
# Test with long string (over 100 chars)
|
|
long_str = "a" * 120
|
|
assert len(clean_filename(long_str)) == 100
|
|
|
|
# Test with empty string
|
|
assert clean_filename("") == "untitled"
|
|
|
|
# Test with only special characters
|
|
# Some implementations may return empty string or underscore
|
|
result = clean_filename("!@#$%^&*()")
|
|
assert result in ["untitled", "_", ""]
|
|
|
|
|
|
def test_format_timestamp():
|
|
"""Test format_timestamp utility function."""
|
|
# Test with datetime object
|
|
dt = datetime(2023, 1, 1, 12, 30, 45)
|
|
assert format_timestamp(dt) == "2023-01-01 12:30:45"
|
|
|
|
# Test with ISO format string
|
|
iso_str = "2023-01-01T12:30:45Z"
|
|
assert format_timestamp(iso_str) == "2023-01-01 12:30:45"
|
|
|
|
# Test with Unix timestamp as int
|
|
unix_ts = 1672577445 # 2023-01-01 12:30:45 UTC
|
|
formatted = format_timestamp(unix_ts)
|
|
# The exact format may vary by timezone, so we just check for the year
|
|
assert "2023" in formatted
|
|
|
|
# Test with Unix timestamp as string
|
|
unix_str = "1672577445"
|
|
formatted = format_timestamp(unix_str)
|
|
assert "2023" in formatted
|
|
|
|
# Test with unparseable string
|
|
assert format_timestamp("not a timestamp") == "not a timestamp"
|
|
|
|
# Test with non-timestamp object
|
|
assert format_timestamp(None) == "None"
|