ultralytics-actions 0.0.81__tar.gz → 0.0.83__tar.gz

{ultralytics_actions-0.0.81 → ultralytics_actions-0.0.83}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-actions
-Version: 0.0.81
+Version: 0.0.83
 Summary: Ultralytics Actions for GitHub automation and PR management.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>

{ultralytics_actions-0.0.81 → ultralytics_actions-0.0.83}/actions/__init__.py

@@ -22,4 +22,4 @@
 #     ├── test_summarize_pr.py
 #     └── ...
 
-__version__ = "0.0.81"
+__version__ = "0.0.83"

{ultralytics_actions-0.0.81 → ultralytics_actions-0.0.83}/actions/update_file_headers.py

@@ -39,8 +39,6 @@ COMMENT_MAP = {
     # HTML/XML style
     ".html": (None, "<!-- ", " -->"),
     ".xml": (None, "<!-- ", " -->"),  # Android XML
-    # MATLAB style
-    ".m": ("% ", None, None),
 }
 
 # Ignore these Paths (do not update their headers)

{ultralytics_actions-0.0.81 → ultralytics_actions-0.0.83}/actions/utils/github_utils.py

@@ -115,7 +115,12 @@ class Action:
         """Retrieves the diff content for a specified pull request."""
         url = f"{GITHUB_API_URL}/repos/{self.repository}/pulls/{self.pr.get('number')}"
         response = self.get(url, headers=self.headers_diff)
-        return response.text if response.status_code == 200 else ""
+        if response.status_code == 200:
+            return response.text
+        elif response.status_code == 406:
+            return "**ERROR: DIFF TOO LARGE - PR exceeds GitHub's 20,000 line limit, unable to retrieve diff."
+        else:
+            return "**ERROR: UNABLE TO RETRIEVE DIFF."
 
     def get_repo_data(self, endpoint: str) -> dict:
         """Fetches repository data from a specified endpoint."""

ultralytics_actions-0.0.83/tests/test_file_headers.py

@@ -0,0 +1,114 @@
+# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
+
+# tests/test_update_file_headers.py
+"""Tests for the file headers update functionality."""
+
+from pathlib import Path
+from tempfile import TemporaryDirectory
+
+from actions.update_file_headers import COMMENT_MAP, IGNORE_PATHS, update_file
+
+
+def test_update_file_python():
+    """Test updating Python file headers."""
+    with TemporaryDirectory() as tmp_dir:
+        # Create a test Python file
+        test_file = Path(tmp_dir) / "test.py"
+        test_file.write_text("print('Hello World')\n")
+
+        # Update file
+        result = update_file(test_file, "# ", None, None, "Ultralytics 🚀 Test Header")
+
+        # Check results
+        assert result is True
+        content = test_file.read_text()
+        assert content.startswith("# Ultralytics 🚀 Test Header\n\n")
+        assert "print('Hello World')" in content
+
+
+def test_update_file_cpp():
+    """Test updating C++ file headers."""
+    with TemporaryDirectory() as tmp_dir:
+        # Create a test C++ file
+        test_file = Path(tmp_dir) / "test.cpp"
+        test_file.write_text("#include <iostream>\n\nint main() {\n    return 0;\n}\n")
+
+        # Update file
+        result = update_file(test_file, "// ", "/* ", " */", "Ultralytics 🚀 Test Header")
+
+        # Check results
+        assert result is True
+        content = test_file.read_text()
+        assert content.startswith("// Ultralytics 🚀 Test Header\n\n")
+        assert "#include <iostream>" in content
+
+
+def test_update_file_with_existing_header():
+    """Test updating file with existing header."""
+    with TemporaryDirectory() as tmp_dir:
+        # Create a test file with existing header
+        test_file = Path(tmp_dir) / "test.py"
+        test_file.write_text("# Ultralytics 🚀 AGPL-3.0 License\n\ndef main():\n    pass\n")
+
+        # Update file
+        result = update_file(test_file, "# ", None, None, "Ultralytics 🚀 Test Header")
+
+        # Check results
+        assert result is True
+        content = test_file.read_text()
+        assert content.startswith("# Ultralytics 🚀 Test Header\n\n")
+        assert "def main():" in content
+
+
+def test_update_file_with_shebang():
+    """Test updating file with shebang."""
+    with TemporaryDirectory() as tmp_dir:
+        # Create a test file with shebang
+        test_file = Path(tmp_dir) / "test.py"
+        test_file.write_text("#!/usr/bin/env python3\n\nprint('Hello World')\n")
+
+        # Update file
+        result = update_file(test_file, "# ", None, None, "Ultralytics 🚀 Test Header")
+
+        # Check results
+        assert result is True
+        content = test_file.read_text()
+        assert content.startswith("#!/usr/bin/env python3\n# Ultralytics 🚀 Test Header\n\n")
+        assert "print('Hello World')" in content
+
+
+def test_update_file_no_changes():
+    """Test updating file with no changes needed."""
+    with TemporaryDirectory() as tmp_dir:
+        # Create a test file with correct header
+        test_file = Path(tmp_dir) / "test.py"
+        test_file.write_text("# Ultralytics 🚀 Test Header\n\nprint('Hello World')\n")
+
+        # Update file
+        result = update_file(test_file, "# ", None, None, "Ultralytics 🚀 Test Header")
+
+        # Check results
+        assert result is False  # No changes made
+
+
+def test_comment_map_coverage():
+    """Test that all supported file extensions have defined comment styles."""
+    # Check for a few key extensions
+    assert ".py" in COMMENT_MAP
+    assert ".cpp" in COMMENT_MAP
+    assert ".js" in COMMENT_MAP
+    assert ".html" in COMMENT_MAP
+
+    # Check comment style format
+    for ext, (prefix, block_start, block_end) in COMMENT_MAP.items():
+        assert isinstance(ext, str)
+        assert prefix is None or isinstance(prefix, str)
+        assert block_start is None or isinstance(block_start, str)
+        assert block_end is None or isinstance(block_end, str)
+
+
+def test_ignore_paths():
+    """Test that the ignore paths list exists and contains expected entries."""
+    assert isinstance(IGNORE_PATHS, set)
+    assert ".git" in IGNORE_PATHS
+    assert "__pycache__" in IGNORE_PATHS

{ultralytics_actions-0.0.81 → ultralytics_actions-0.0.83}/ultralytics_actions.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultralytics-actions
-Version: 0.0.81
+Version: 0.0.83
 Summary: Ultralytics Actions for GitHub automation and PR management.
 Author-email: Glenn Jocher <glenn.jocher@ultralytics.com>
 Maintainer-email: Ultralytics <hello@ultralytics.com>

{ultralytics_actions-0.0.81 → ultralytics_actions-0.0.83}/ultralytics_actions.egg-info/SOURCES.txt

@@ -15,6 +15,7 @@ actions/utils/openai_utils.py
 tests/test_cli_commands.py
 tests/test_common_utils.py
 tests/test_dispatch_actions.py
+tests/test_file_headers.py
 tests/test_first_interaction.py
 tests/test_github_utils.py
 tests/test_init.py