gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: format file

Browse source
This commit is contained in:
Igor Ilic 2025-06-30 12:26:43 +02:00
parent d4d12ab7b7
commit 2c7eecc93d
1 changed files with 19 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
cognee/tests/unit/modules/data/test_open_data_file.py
View file

@ -9,13 +9,13 @@ class TestOpenDataFile:
def test_regular_file_path(self):
"""Test that regular file paths work as before."""
with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.txt') as f:
with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".txt") as f:
test_content = "Test content for regular file path"
f.write(test_content)
temp_file_path = f.name
try:
with open_data_file(temp_file_path, mode='r') as f:
with open_data_file(temp_file_path, mode="r") as f:
content = f.read()
assert content == test_content
finally:
@ -23,14 +23,14 @@ class TestOpenDataFile:
def test_file_url_text_mode(self):
"""Test that file:// URLs work correctly in text mode."""
with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.txt') as f:
with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".txt") as f:
test_content = "Test content for file:// URL handling"
f.write(test_content)
temp_file_path = f.name
try:
file_url = f"file://{temp_file_path}"
with open_data_file(file_url, mode='r') as f:
with open_data_file(file_url, mode="r") as f:
content = f.read()
assert content == test_content
finally:
@ -38,14 +38,14 @@ class TestOpenDataFile:
def test_file_url_binary_mode(self):
"""Test that file:// URLs work correctly in binary mode."""
with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.txt') as f:
with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".txt") as f:
test_content = "Test content for binary mode"
f.write(test_content)
temp_file_path = f.name
try:
file_url = f"file://{temp_file_path}"
with open_data_file(file_url, mode='rb') as f:
with open_data_file(file_url, mode="rb") as f:
content = f.read()
assert content == test_content.encode()
finally:
@ -53,14 +53,16 @@ class TestOpenDataFile:
def test_file_url_with_encoding(self):
"""Test that file:// URLs work with specific encoding."""
with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.txt', encoding='utf-8') as f:
with tempfile.NamedTemporaryFile(
mode="w", delete=False, suffix=".txt", encoding="utf-8"
) as f:
test_content = "Test content with UTF-8: café ☕"
f.write(test_content)
temp_file_path = f.name
try:
file_url = f"file://{temp_file_path}"
with open_data_file(file_url, mode='r', encoding='utf-8') as f:
with open_data_file(file_url, mode="r", encoding="utf-8") as f:
content = f.read()
assert content == test_content
finally:
@ -70,20 +72,20 @@ class TestOpenDataFile:
"""Test that file:// URLs raise appropriate error for nonexistent files."""
file_url = "file:///nonexistent/path/to/file.txt"
with pytest.raises(FileNotFoundError):
with open_data_file(file_url, mode='r') as f:
with open_data_file(file_url, mode="r") as f:
f.read()
def test_multiple_file_prefixes(self):
"""Test that multiple file:// prefixes are handled correctly."""
with tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.txt') as f:
with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".txt") as f:
test_content = "Test content"
f.write(test_content)
temp_file_path = f.name
try:
# Even if someone accidentally adds multiple file:// prefixes
file_url = f"file://file://{temp_file_path}"
with open_data_file(file_url, mode='r') as f:
with open_data_file(file_url, mode="r") as f:
content = f.read()
# This should work because we only replace the first occurrence
assert content == test_content
@ -95,4 +97,4 @@ class TestOpenDataFile:
if __name__ == "__main__":
pytest.main([__file__, "-v"])
pytest.main([__file__, "-v"])