⚡️ Speed up function ask_for_telemetry by 14% in PR #186 (cf-621)

[codeflash-ai]

⚡️ This pull request contains optimizations for PR #186

If you approve this dependent PR, these changes will be merged into the original PR branch cf-621.

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

---

📄 14% (0.14x) speedup for ask_for_telemetry in codeflash/cli_cmds/cmd_init.py

⏱️ Runtime : 5.81 milliseconds → 5.08 milliseconds (best of 372 runs)

📝 Explanation and details

Here is an optimized version of your program.

Optimization applied:

• The from rich.prompt import Confirm import has been moved to the top level.
  • This avoids re-importing the module every time ask_for_telemetry() is called, which saves unnecessary import overhead and speeds up repeated calls.

✅ Correctness verification report:

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

🌀 Generated Regression Tests Details

from __future__ import annotations

from unittest.mock import patch

# imports
import pytest  # used for our unit tests
from codeflash.cli_cmds.cmd_init import ask_for_telemetry
from rich.prompt import Confirm

# unit tests

def test_ask_for_telemetry_yes():
    """Test when user inputs 'yes'."""
    with patch('rich.prompt.Confirm.ask', return_value=True):
        codeflash_output = ask_for_telemetry()

def test_ask_for_telemetry_no():
    """Test when user inputs 'no'."""
    with patch('rich.prompt.Confirm.ask', return_value=False):
        codeflash_output = ask_for_telemetry()

def test_ask_for_telemetry_default():
    """Test default behavior when user presses Enter."""
    with patch('rich.prompt.Confirm.ask', return_value=True):
        codeflash_output = ask_for_telemetry()

def test_ask_for_telemetry_case_insensitivity():
    """Test case insensitivity for 'yes' and 'no' inputs."""
    with patch('rich.prompt.Confirm.ask', side_effect=['Yes', 'YES', 'yEs']):
        codeflash_output = ask_for_telemetry()
    with patch('rich.prompt.Confirm.ask', side_effect=['No', 'NO', 'nO']):
        codeflash_output = ask_for_telemetry()

def test_ask_for_telemetry_unexpected_input():
    """Test handling of unexpected input."""
    with patch('rich.prompt.Confirm.ask', side_effect=['abc', '123']):
        codeflash_output = ask_for_telemetry()  # Assuming unexpected inputs default to 'no'

def test_ask_for_telemetry_special_characters():
    """Test prompt message with special characters."""
    with patch('rich.prompt.Confirm.ask', return_value=True) as mock_confirm:
        ask_for_telemetry()
        mock_confirm.assert_called_with(
            "⚡️ Would you like to enable telemetry to help us improve the Codeflash experience?",
            default=True,
            show_default=True
        )

def test_ask_for_telemetry_terminal_environment():
    """Test behavior in different terminal environments."""
    with patch('rich.prompt.Confirm.ask', return_value=True):
        codeflash_output = ask_for_telemetry()

def test_ask_for_telemetry_performance():
    """Test performance over repeated calls."""
    with patch('rich.prompt.Confirm.ask', return_value=True):
        for _ in range(1000):  # Simulate repeated calls
            codeflash_output = ask_for_telemetry()
# 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

from unittest.mock import patch  # used to simulate user input

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

# unit tests

def test_default_behavior_accept():
    """Test that the function returns True when the user accepts the default prompt."""
    with patch('rich.prompt.Confirm.ask', return_value=True):
        codeflash_output = ask_for_telemetry()

def test_default_behavior_decline():
    """Test that the function returns False when the user explicitly declines."""
    with patch('rich.prompt.Confirm.ask', return_value=False):
        codeflash_output = ask_for_telemetry()

def test_user_input_variations_yes():
    """Test that the function returns True for various 'yes' inputs."""
    with patch('rich.prompt.Confirm.ask', return_value=True):
        codeflash_output = ask_for_telemetry()

def test_user_input_variations_no():
    """Test that the function returns False for various 'no' inputs."""
    with patch('rich.prompt.Confirm.ask', return_value=False):
        codeflash_output = ask_for_telemetry()

def test_invalid_input():
    """Test that the function handles invalid inputs gracefully."""
    with patch('rich.prompt.Confirm.ask', side_effect=[None, True]):
        codeflash_output = ask_for_telemetry()

def test_edge_case_special_characters():
    """Test that the function handles special characters in input."""
    with patch('rich.prompt.Confirm.ask', return_value=False):
        codeflash_output = ask_for_telemetry()

def test_environment_variations():
    """Test that the function works in different terminal environments."""
    # This test is more conceptual, as actual terminal differences are hard to simulate in unit tests.
    with patch('rich.prompt.Confirm.ask', return_value=True):
        codeflash_output = ask_for_telemetry()

def test_performance():
    """Test the function's performance under simulated high latency."""
    with patch('rich.prompt.Confirm.ask', return_value=True):
        codeflash_output = ask_for_telemetry()
# 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-pr186-2025-05-07T20.42.23 and push.

[codeflash-ai]

This PR has been automatically closed because the original PR #186 by aseembits93 was closed.