⚡️ Speed up ConsoleOptions.reset_height() by 13% in rich/console.py

[codeflash-ai]

📄 ConsoleOptions.reset_height() in rich/console.py

📈 Performance improved by 13% (0.13x faster)

⏱️ Runtime went down from 6.10 microseconds to 5.40 microseconds

Explanation and details

Here is an optimized version of the given Python program. The optimizations include directly updating the attributes instead of creating a new dictionary, which eliminates unnecessary dictionary operations and improves runtime performance.

By iterating through self.__dict__.items() and using setattr to assign the attributes to the new instance, the new version avoids the overhead of the dict.copy method. This approach significantly minimizes memory operations and improves the overall execution speed without altering the function signatures or return values.

Correctness verification

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

🔘 (none found) − ⚙️ Existing Unit Tests

✅ 1 Passed − 🌀 Generated Regression Tests

(click to show generated tests)

# imports
import pytest  # used for our unit tests

# unit tests

def test_reset_height_basic():
    # Test resetting height to None
    options = ConsoleOptions()
    options.height = 10
    new_options = options.reset_height()
    assert new_options.height is None

def test_reset_height_already_none():
    # Test when height is already None
    options = ConsoleOptions()
    options.height = None
    new_options = options.reset_height()
    assert new_options.height is None

def test_reset_height_no_height_attribute():
    # Test when height attribute does not exist
    options = ConsoleOptions()
    new_options = options.reset_height()
    assert hasattr(new_options, 'height')
    assert new_options.height is None

def test_reset_height_other_attributes():
    # Test that other attributes remain unchanged
    options = ConsoleOptions()
    options.width = 20
    options.height = 10
    new_options = options.reset_height()
    assert new_options.height is None
    assert new_options.width == 20

def test_reset_height_different_types():
    # Test with height as different types
    options = ConsoleOptions()
    
    options.height = 10
    new_options = options.reset_height()
    assert new_options.height is None

    options.height = 10.5
    new_options = options.reset_height()
    assert new_options.height is None

    options.height = "tall"
    new_options = options.reset_height()
    assert new_options.height is None

    options.height = [1, 2, 3]
    new_options = options.reset_height()
    assert new_options.height is None

def test_reset_height_immutability():
    # Test that the original instance remains unchanged
    options = ConsoleOptions()
    options.height = 10
    new_options = options.reset_height()
    assert options.height == 10
    assert new_options.height is None

def test_reset_height_inheritance():
    # Test with a subclass of ConsoleOptions
    class AdvancedConsoleOptions(ConsoleOptions):
        def __init__(self):
            self.depth = 30

    options = AdvancedConsoleOptions()
    options.height = 10
    new_options = options.reset_height()
    assert new_options.height is None
    assert new_options.depth == 30

def test_reset_height_large_number_of_attributes():
    # Test with a large number of attributes
    options = ConsoleOptions()
    for i in range(1000):
        setattr(options, f'attr_{i}', i)
    new_options = options.reset_height()
    assert new_options.height is None
    for i in range(1000):
        assert getattr(new_options, f'attr_{i}') == i

def test_reset_height_nested_structures():
    # Test with nested structures
    class NestedOptions:
        def __init__(self):
            self.depth = 30

    options = ConsoleOptions()
    options.height = 10
    options.nested = NestedOptions()
    new_options = options.reset_height()
    assert new_options.height is None
    assert new_options.nested.depth == 30

def test_reset_height_deterministic_behavior():
    # Test that multiple calls produce consistent results
    options = ConsoleOptions()
    options.height = 10
    new_options_1 = options.reset_height()
    new_options_2 = options.reset_height()
    assert new_options_1.height == new_options_2.height == None

def test_reset_height_stress_testing():
    # Stress test with a large number of instances
    for _ in range(10000):
        options = ConsoleOptions()
        options.height = 10
        new_options = options.reset_height()
        assert new_options.height is None

🔘 (none found) − ⏪ Replay Tests