⚡️ Speed up function get_dependency_manager_installation_string by 127% in PR #208 (bump-gha-uv-version)

[codeflash-ai]

⚡️ This pull request contains optimizations for PR #208

If you approve this dependent PR, these changes will be merged into the original PR branch bump-gha-uv-version.

This PR will be automatically closed if the original PR is merged.

---

📄 127% (1.27x) speedup for get_dependency_manager_installation_string in codeflash/cli_cmds/cmd_init.py

⏱️ Runtime : 1.20 millisecond → 529 microseconds (best of 182 runs)

📝 Explanation and details

Here's an optimized version of your program with reduced runtime, specifically targeting the bottlenecks.

Optimizations performed:

1. Constant Folding: The sys.version_info fetch and string formatting for the Python version is only necessary for the non-UV case and does not need to be constructed unless used.
2. Precomputed Templates: The output strings are constants and can be stored as such, to avoid reconstructing them on each call.
3. Avoid Unnecessary Formatting: For the setup-python path, the version string is constant across invocations of the same interpreter, so can use lazy-static initialization.
4. Reduced Function Overhead: Restructured code to minimize code execution paths and avoid unnecessary work.

---

---

Summary of changes:

• Only creates formatted strings once per interpreter lifetime, so on repeated heavy calls, time and allocations are minimized.
• sys.version_info is not re-accessed on every call.
• Preserved all logical comments; comments were added only where code was optimized.

This rewrite should significantly improve per-call performance of get_dependency_manager_installation_string().

✅ Correctness verification report:

Test	Status
⚙️ Existing Unit Tests	🔘 None Found
🌀 Generated Regression Tests	✅ 2334 Passed
⏪ Replay Tests	🔘 None Found
🔎 Concolic Coverage Tests	🔘 None Found
📊 Tests Coverage

🌀 Generated Regression Tests Details

from __future__ import annotations

import sys
from enum import Enum

# imports
import pytest  # used for our unit tests
from codeflash.cli_cmds.cmd_init import \
    get_dependency_manager_installation_string


class DependencyManager(Enum):
    PIP = "pip"
    POETRY = "poetry"
    PDM = "pdm"
    UV = "uv"
    CUSTOM = "custom"
from codeflash.cli_cmds.cmd_init import \
    get_dependency_manager_installation_string

# unit tests

# -------------------------------
# Basic Test Cases
# -------------------------------

def test_install_string_for_pip():
    """Test that the installation string for PIP is correct and contains the expected Python version."""
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.PIP); result = codeflash_output
    py_version = sys.version_info
    expected_version = f"'{py_version.major}.{py_version.minor}'"

def test_install_string_for_poetry():
    """Test that the installation string for Poetry is correct and contains the expected Python version."""
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.POETRY); result = codeflash_output
    py_version = sys.version_info
    expected_version = f"'{py_version.major}.{py_version.minor}'"

def test_install_string_for_pdm():
    """Test that the installation string for PDM is correct and contains the expected Python version."""
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.PDM); result = codeflash_output
    py_version = sys.version_info
    expected_version = f"'{py_version.major}.{py_version.minor}'"

def test_install_string_for_custom():
    """Test that the installation string for CUSTOM is correct and contains the expected Python version."""
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.CUSTOM); result = codeflash_output
    py_version = sys.version_info
    expected_version = f"'{py_version.major}.{py_version.minor}'"

def test_install_string_for_uv():
    """Test that the installation string for UV is correct and does not include python-version."""
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.UV); result = codeflash_output

# -------------------------------
# Edge Test Cases
# -------------------------------




def test_python_version_format():
    """Test that the python version is always quoted and in the correct format."""
    for manager in [DependencyManager.PIP, DependencyManager.POETRY, DependencyManager.PDM, DependencyManager.CUSTOM]:
        codeflash_output = get_dependency_manager_installation_string(manager); result = codeflash_output
        py_version = sys.version_info
        expected_version = f"'{py_version.major}.{py_version.minor}'"

def test_uv_string_exact_match():
    """Test that the UV installation string matches exactly (no python-version)."""
    expected = """name: 🐍 Setup UV
        uses: astral-sh/setup-uv@v6
        with:
          enable-cache: true"""
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.UV); result = codeflash_output

def test_setup_python_string_exact_match():
    """Test that the setup-python string matches exactly for a given manager."""
    py_version = sys.version_info
    expected = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.PIP); result = codeflash_output

# -------------------------------
# Large Scale Test Cases
# -------------------------------

@pytest.mark.parametrize("manager", [DependencyManager.PIP, DependencyManager.POETRY, DependencyManager.PDM, DependencyManager.CUSTOM])
def test_bulk_managers_return_unique_strings(manager):
    """Test that each manager (except UV) returns the same setup-python string."""
    py_version = sys.version_info
    expected = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""
    for _ in range(250):  # Test in bulk for performance and determinism
        codeflash_output = get_dependency_manager_installation_string(manager); result = codeflash_output

def test_all_enum_members():
    """Test that all enum members are handled and produce the expected output."""
    py_version = sys.version_info
    expected_setup = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""
    expected_uv = """name: 🐍 Setup UV
        uses: astral-sh/setup-uv@v6
        with:
          enable-cache: true"""
    for manager in DependencyManager:
        codeflash_output = get_dependency_manager_installation_string(manager); result = codeflash_output
        if manager == DependencyManager.UV:
            pass
        else:
            pass

def test_repeated_calls_are_consistent():
    """Test that repeated calls with the same input give the same output."""
    for manager in DependencyManager:
        results = [get_dependency_manager_installation_string(manager) for _ in range(100)]
        first = results[0]
        for r in results:
            pass

def test_large_batch_of_managers():
    """Test a large batch of manager inputs for performance and consistency."""
    managers = [DependencyManager.PIP, DependencyManager.POETRY, DependencyManager.PDM, DependencyManager.CUSTOM, DependencyManager.UV]
    # Create a list of 500 managers, cycling through the enum
    batch = [managers[i % len(managers)] for i in range(500)]
    py_version = sys.version_info
    expected_setup = f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""
    expected_uv = """name: 🐍 Setup UV
        uses: astral-sh/setup-uv@v6
        with:
          enable-cache: true"""
    for i, manager in enumerate(batch):
        codeflash_output = get_dependency_manager_installation_string(manager); result = codeflash_output
        if manager == DependencyManager.UV:
            pass
        else:
            pass
# codeflash_output is used to check that the output of the original code is the same as that of the optimized code.

from __future__ import annotations

import sys
from enum import Enum

# imports
import pytest  # used for our unit tests
from codeflash.cli_cmds.cmd_init import \
    get_dependency_manager_installation_string


class DependencyManager(Enum):
    PIP = "pip"
    POETRY = "poetry"
    PDM = "pdm"
    UV = "uv"
from codeflash.cli_cmds.cmd_init import \
    get_dependency_manager_installation_string

# unit tests

# ------------------------- #
# 1. Basic Test Cases       #
# ------------------------- #

def test_pip_manager_returns_python_setup_string():
    # Test basic case for PIP dependency manager
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.PIP); result = codeflash_output
    py_version = sys.version_info
    expected = (
        f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""
    )

def test_poetry_manager_returns_python_setup_string():
    # Test basic case for POETRY dependency manager
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.POETRY); result = codeflash_output
    py_version = sys.version_info
    expected = (
        f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""
    )

def test_pdm_manager_returns_python_setup_string():
    # Test basic case for PDM dependency manager
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.PDM); result = codeflash_output
    py_version = sys.version_info
    expected = (
        f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""
    )

def test_uv_manager_returns_uv_setup_string():
    # Test basic case for UV dependency manager
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.UV); result = codeflash_output
    expected = """name: 🐍 Setup UV
        uses: astral-sh/setup-uv@v6
        with:
          enable-cache: true"""

# ------------------------- #
# 2. Edge Test Cases        #
# ------------------------- #




def test_python_version_string_formatting():
    # Test that the python version string is always formatted as 'major.minor'
    for dep in [DependencyManager.PIP, DependencyManager.POETRY, DependencyManager.PDM]:
        codeflash_output = get_dependency_manager_installation_string(dep); result = codeflash_output
        py_version = sys.version_info
        expected_version = f"'{py_version.major}.{py_version.minor}'"


def test_all_enum_members_in_large_list():
    # Test the function with a large list of DependencyManager members to ensure no state leaks and performance is reasonable
    members = [DependencyManager.PIP, DependencyManager.POETRY, DependencyManager.PDM, DependencyManager.UV] * 200  # 800 items
    py_version = sys.version_info
    expected_python = (
        f"""name: 🐍 Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '{py_version.major}.{py_version.minor}'"""
    )
    expected_uv = """name: 🐍 Setup UV
        uses: astral-sh/setup-uv@v6
        with:
          enable-cache: true"""
    # Test all outputs in a loop
    for i, dep in enumerate(members):
        codeflash_output = get_dependency_manager_installation_string(dep); result = codeflash_output
        if dep == DependencyManager.UV:
            pass
        else:
            pass


def test_unique_output_for_each_manager():
    # Test that UV and non-UV managers produce different outputs
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.UV); uv_output = codeflash_output
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.PIP); pip_output = codeflash_output
    # Check that all non-UV managers produce the same output
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.POETRY); poetry_output = codeflash_output
    codeflash_output = get_dependency_manager_installation_string(DependencyManager.PDM); pdm_output = codeflash_output
# codeflash_output is used to check that the output of the original code is the same as that of the optimized code.

To edit these changes git checkout codeflash/optimize-pr208-2025-05-15T21.13.02 and push.