⚡️ Speed up function parse_req_python_version by 8% in src/black/files.py

[codeflash-ai]

📄 parse_req_python_version in src/black/files.py

✨ Performance Summary:

• Speed Increase: 📈 8% (0.08x faster)
• Runtime Reduction: ⏱️ From 253 microseconds down to 235 microseconds (best of 108 runs)

---

📝 Explanation and details

Here's a potential optimized version of your Python program. I've made a few adjustments to the original code to improve its performance.

1. I've reduced the number of Enum class members by using a more concise range definition.
2. I've optimized the pretty method to leverage string manipulation better.
3. I've streamlined the parse_req_python_version function to avoid unnecessary exceptions.

Key changes.

1. The pretty method now checks the length of the name once, rather than slicing it twice.
2. The parse_req_python_version function now directly checks whether the major version is 3 and whether there are at least two elements in release to prevent IndexError. It also uses simpler logic for version range checks.

The logic remains unchanged, ensuring the same output while potentially improving performance.

---

✅ Correctness verification

The new optimized code was tested for correctness. The results are listed below:

Test	Status	Details
⚙️ Existing Unit Tests	🔘 None Found
🌀 Generated Regression Tests	✅ 40 Passed	See below
⏪ Replay Tests	🔘 None Found
🔎 Concolic Coverage Tests	🔘 None Found
📊 Coverage	100.0%

🌀 Generated Regression Tests Details

Click to view details

from enum import Enum
from typing import Optional

# imports
import pytest  # used for our unit tests
from black.files import parse_req_python_version
# function to test
from packaging.version import InvalidVersion, Version

# unit tests

# Basic Test Cases
def test_valid_python_versions():
    codeflash_output = parse_req_python_version("3.3")
    codeflash_output = parse_req_python_version("3.7")
    codeflash_output = parse_req_python_version("3.10")
    codeflash_output = parse_req_python_version("3.12")

def test_invalid_major_versions():
    codeflash_output = parse_req_python_version("2.7")
    codeflash_output = parse_req_python_version("4.0")
    codeflash_output = parse_req_python_version("1.5")

def test_invalid_minor_versions():
    codeflash_output = parse_req_python_version("3.2")
    codeflash_output = parse_req_python_version("3.14")
    codeflash_output = parse_req_python_version("3.0")


def test_empty_string():
    with pytest.raises(InvalidVersion):
        parse_req_python_version("")

def test_none_input():
    with pytest.raises(TypeError):
        parse_req_python_version(None)




def test_non_string_inputs():
    with pytest.raises(TypeError):
        parse_req_python_version(3.7)
    with pytest.raises(TypeError):
        parse_req_python_version(37)
    with pytest.raises(TypeError):
        parse_req_python_version(["3.7"])
    with pytest.raises(TypeError):
        parse_req_python_version(None)


def test_version_strings_with_mixed_case():
    with pytest.raises(InvalidVersion):
        parse_req_python_version("Python3.7")
    with pytest.raises(InvalidVersion):
        parse_req_python_version("PYTHON3.7")
    with pytest.raises(InvalidVersion):
        parse_req_python_version("python3.7")

def test_version_strings_with_non_ascii_characters():
    with pytest.raises(InvalidVersion):
        parse_req_python_version("3.7é")
    with pytest.raises(InvalidVersion):
        parse_req_python_version("3.7π")
    with pytest.raises(InvalidVersion):
        parse_req_python_version("3.7你")

# Large Scale Test Cases


def test_boundary_versions():
    codeflash_output = parse_req_python_version("3.3")
    codeflash_output = parse_req_python_version("3.13")
    codeflash_output = parse_req_python_version("3.2")
    codeflash_output = parse_req_python_version("3.14")
# codeflash_output is used to check that the output of the original code is the same as that of the optimized code.
from enum import Enum
from typing import Optional

# imports
import pytest  # used for our unit tests
from black.files import parse_req_python_version
# function to test
from packaging.version import InvalidVersion, Version

# unit tests

# Basic Test Cases
def test_valid_version_3_7():
    codeflash_output = parse_req_python_version("3.7")

def test_valid_version_3_6():
    codeflash_output = parse_req_python_version("3.6")

def test_boundary_valid_version_3_3():
    codeflash_output = parse_req_python_version("3.3")

def test_boundary_valid_version_3_13():
    codeflash_output = parse_req_python_version("3.13")

# Invalid Inputs
def test_invalid_major_version_2_7():
    codeflash_output = parse_req_python_version("2.7")

def test_invalid_major_version_4_0():
    codeflash_output = parse_req_python_version("4.0")

def test_invalid_minor_version_3_20():
    codeflash_output = parse_req_python_version("3.20")


def test_non_numeric_version_three_seven():
    with pytest.raises(InvalidVersion):
        parse_req_python_version("three.seven")

def test_non_numeric_version_3_seven():
    with pytest.raises(InvalidVersion):
        parse_req_python_version("3.seven")


def test_incomplete_version_3_dot():
    with pytest.raises(InvalidVersion):
        parse_req_python_version("3.")

# Edge Cases
def test_empty_string():
    with pytest.raises(InvalidVersion):
        parse_req_python_version("")

def test_whitespace_string():
    with pytest.raises(InvalidVersion):
        parse_req_python_version(" ")

def test_leading_trailing_whitespace():
    codeflash_output = parse_req_python_version(" 3.7 ")

---

📣 **Feedback**

If you have any feedback or need assistance, feel free to join our Discord community: