⚡️ Speed up function ask_for_telemetry by 17% in PR #184 (cache_discovered_tests)

[codeflash-ai]

⚡️ This pull request contains optimizations for PR #184

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

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

---

📄 17% (0.17x) speedup for ask_for_telemetry in codeflash/cli_cmds/cmd_init.py

⏱️ Runtime : 105 microseconds → 89.6 microseconds (best of 132 runs)

📝 Explanation and details

✅ Correctness verification report:

Test	Status
⚙️ Existing Unit Tests	🔘 None Found
🌀 Generated Regression Tests	✅ 7 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  # used to mock the Confirm.ask method

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

# unit tests

# Test case for basic functionality with user confirming
def test_confirm_yes():
    with patch('rich.prompt.Confirm.ask', return_value=True):
        codeflash_output = ask_for_telemetry()

# Test case for basic functionality with user denying
def test_confirm_no():
    with patch('rich.prompt.Confirm.ask', return_value=False):
        codeflash_output = ask_for_telemetry()

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

# Test case for invalid input handling

def test_non_interactive_environment():
    with patch('rich.prompt.Confirm.ask', side_effect=EOFError("No input available")):
        with pytest.raises(EOFError):
            ask_for_telemetry()

# Test case for interrupted input
def test_interrupted_input():
    with patch('rich.prompt.Confirm.ask', side_effect=KeyboardInterrupt):
        with pytest.raises(KeyboardInterrupt):
            ask_for_telemetry()

# Test case for rich library not installed

def test_terminal_redirection():
    with patch('rich.prompt.Confirm.ask', side_effect=EOFError("Input redirected")):
        with pytest.raises(EOFError):
            ask_for_telemetry()

# Test case for resource constraints
def test_resource_constraints():
    # This test is more conceptual as simulating low resources is complex.
    # We can simulate by assuming Confirm.ask fails due to resource limits.
    with patch('rich.prompt.Confirm.ask', side_effect=MemoryError("Out of memory")):
        with pytest.raises(MemoryError):
            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-pr184-2025-05-06T23.48.47 and push.

[codeflash-ai]

This PR has been automatically closed because the original PR #184 by KRRT7 was closed.