From 2c7eecc93d94a142a573b47289693bf108245194 Mon Sep 17 00:00:00 2001
From: Igor Ilic <igorilic03@gmail.com>
Date: Mon, 30 Jun 2025 12:26:43 +0200
Subject: [PATCH] refactor: format file

---
 .../unit/modules/data/test_open_data_file.py  | 36 ++++++++++---------
 1 file changed, 19 insertions(+), 17 deletions(-)

diff --git a/cognee/tests/unit/modules/data/test_open_data_file.py b/cognee/tests/unit/modules/data/test_open_data_file.py
index 5be04a1c3..eea402aa3 100644
--- a/cognee/tests/unit/modules/data/test_open_data_file.py
+++ b/cognee/tests/unit/modules/data/test_open_data_file.py
@@ -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"])
\ No newline at end of file
+    pytest.main([__file__, "-v"])