⚡️ Speed up method Renderer.ax_zoomable by 7%

[codeflash-ai]

📄 7% (0.07x) speedup for Renderer.ax_zoomable in plotly/matplotlylib/mplexporter/renderers/base.py

⏱️ Runtime : 412 microseconds → 385 microseconds (best of 335 runs)

📝 Explanation and details

Here's an optimized version of your program. Since the logic is simple and short, only minimal optimizations are possible.

• Avoid the unnecessary bool() conversion: ax.get_navigate() already returns a Boolean.
• Use early return for clarity and efficiency.

Optimized code.

There are no expensive operations to optimize further. The only meaningful improvement was to avoid unnecessary calls and keep logic concise.

✅ Correctness verification report:

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

🌀 Generated Regression Tests Details

import pytest  # used for our unit tests
from plotly.matplotlylib.mplexporter.renderers.base import Renderer


# Mock class to simulate matplotlib Axes behavior for testing
class MockAxes:
    def __init__(self, navigate=True):
        self._navigate = navigate
    def get_navigate(self):
        return self._navigate

# unit tests

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

def test_zoomable_true():
    """Test with an axes object that is zoomable (navigate=True)."""
    ax = MockAxes(navigate=True)
    codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_false():
    """Test with an axes object that is not zoomable (navigate=False)."""
    ax = MockAxes(navigate=False)
    codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_explicit_none():
    """Test with None as input (should return False)."""
    codeflash_output = Renderer.ax_zoomable(None)

def test_zoomable_explicit_false():
    """Test with False as input (should return False)."""
    codeflash_output = Renderer.ax_zoomable(False)

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

def test_zoomable_object_without_get_navigate():
    """Test with an object that does not have get_navigate method (should raise AttributeError)."""
    class NoNavigate:
        pass
    obj = NoNavigate()
    with pytest.raises(AttributeError):
        Renderer.ax_zoomable(obj)

def test_zoomable_object_with_get_navigate_none():
    """Test with an object whose get_navigate returns None (should return False)."""
    class AxWithNone:
        def get_navigate(self):
            return None
    ax = AxWithNone()
    codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_object_with_get_navigate_nonbool():
    """Test with get_navigate returning non-bool values (should coerce to bool)."""
    class AxWithInt:
        def get_navigate(self):
            return 1  # truthy
    ax = AxWithInt()
    codeflash_output = Renderer.ax_zoomable(ax)

    class AxWithZero:
        def get_navigate(self):
            return 0  # falsy
    ax = AxWithZero()
    codeflash_output = Renderer.ax_zoomable(ax)

    class AxWithString:
        def get_navigate(self):
            return "yes"  # truthy
    ax = AxWithString()
    codeflash_output = Renderer.ax_zoomable(ax)

    class AxWithEmptyString:
        def get_navigate(self):
            return ""  # falsy
    ax = AxWithEmptyString()
    codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_object_with_get_navigate_exception():
    """Test with get_navigate raising an exception (should propagate)."""
    class AxWithException:
        def get_navigate(self):
            raise RuntimeError("Navigation error")
    ax = AxWithException()
    with pytest.raises(RuntimeError):
        Renderer.ax_zoomable(ax)

def test_zoomable_object_with_get_navigate_callable():
    """Test with get_navigate returning a callable (should be True as non-None)."""
    class AxWithCallable:
        def get_navigate(self):
            return lambda: True
    ax = AxWithCallable()
    codeflash_output = Renderer.ax_zoomable(ax)

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

def test_zoomable_many_axes_true():
    """Test with a large number of zoomable axes (all should return True)."""
    axes = [MockAxes(navigate=True) for _ in range(500)]
    for ax in axes:
        codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_many_axes_false():
    """Test with a large number of non-zoomable axes (all should return False)."""
    axes = [MockAxes(navigate=False) for _ in range(500)]
    for ax in axes:
        codeflash_output = Renderer.ax_zoomable(ax)



def test_zoomable_with_unusual_truthy_falsy():
    """Test with get_navigate returning unusual truthy/falsy values."""
    class AxWithList:
        def get_navigate(self):
            return [1]  # truthy
    ax = AxWithList()
    codeflash_output = Renderer.ax_zoomable(ax)

    class AxWithEmptyList:
        def get_navigate(self):
            return []  # falsy
    ax = AxWithEmptyList()
    codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_with_inherited_get_navigate():
    """Test with an object inheriting get_navigate from a parent class."""
    class BaseAx:
        def get_navigate(self):
            return True
    class DerivedAx(BaseAx):
        pass
    ax = DerivedAx()
    codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_with_property_get_navigate():
    """Test with get_navigate as a property."""
    class AxWithProperty:
        @property
        def get_navigate(self):
            return lambda: True  # property returns a callable, which is truthy
    ax = AxWithProperty()
    # Calling ax.get_navigate() returns a callable, which is truthy
    codeflash_output = Renderer.ax_zoomable(ax)
# codeflash_output is used to check that the output of the original code is the same as that of the optimized code.

import pytest  # used for our unit tests
from plotly.matplotlylib.mplexporter.renderers.base import Renderer


# Helper class to simulate an axis object
class DummyAx:
    def __init__(self, navigate):
        self._navigate = navigate
    def get_navigate(self):
        return self._navigate


# unit tests

# 1. Basic Test Cases

def test_zoomable_true_with_navigate_true():
    """Test: ax_zoomable returns True when ax.get_navigate() is True."""
    ax = DummyAx(True)
    codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_false_with_navigate_false():
    """Test: ax_zoomable returns False when ax.get_navigate() is False."""
    ax = DummyAx(False)
    codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_false_with_none():
    """Test: ax_zoomable returns False when ax is None."""
    codeflash_output = Renderer.ax_zoomable(None)

def test_zoomable_false_with_empty():
    """Test: ax_zoomable returns False when ax is an empty value (e.g., 0, '')."""
    codeflash_output = Renderer.ax_zoomable(0)
    codeflash_output = Renderer.ax_zoomable('')
    codeflash_output = Renderer.ax_zoomable([])
    codeflash_output = Renderer.ax_zoomable({})

def test_zoomable_true_with_custom_object_true():
    """Test: ax_zoomable returns True with a custom object that returns True from get_navigate."""
    class CustomAx:
        def get_navigate(self):
            return True
    ax = CustomAx()
    codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_false_with_custom_object_false():
    """Test: ax_zoomable returns False with a custom object that returns False from get_navigate."""
    class CustomAx:
        def get_navigate(self):
            return False
    ax = CustomAx()
    codeflash_output = Renderer.ax_zoomable(ax)

# 2. Edge Test Cases

def test_zoomable_raises_if_get_navigate_missing():
    """Test: ax_zoomable raises AttributeError if ax does not have get_navigate()."""
    class NoNavigate:
        pass
    ax = NoNavigate()
    with pytest.raises(AttributeError):
        Renderer.ax_zoomable(ax)

def test_zoomable_with_get_navigate_returning_nonbool():
    """Test: ax_zoomable correctly interprets non-bool get_navigate return values."""
    class AxInt:
        def get_navigate(self):
            return 1  # truthy int
    class AxZero:
        def get_navigate(self):
            return 0  # falsy int
    class AxStr:
        def get_navigate(self):
            return "yes"  # non-empty string, truthy
    class AxEmptyStr:
        def get_navigate(self):
            return ""  # empty string, falsy
    codeflash_output = Renderer.ax_zoomable(AxInt())
    codeflash_output = Renderer.ax_zoomable(AxZero())
    codeflash_output = Renderer.ax_zoomable(AxStr())
    codeflash_output = Renderer.ax_zoomable(AxEmptyStr())

def test_zoomable_with_get_navigate_raises():
    """Test: ax_zoomable propagates exceptions from get_navigate()."""
    class AxRaise:
        def get_navigate(self):
            raise RuntimeError("fail")
    ax = AxRaise()
    with pytest.raises(RuntimeError):
        Renderer.ax_zoomable(ax)

def test_zoomable_with_weird_object():
    """Test: ax_zoomable returns False for objects that are falsy but have get_navigate()."""
    class FalsyButNavigate:
        def __bool__(self):
            return False
        def get_navigate(self):
            return True
    ax = FalsyButNavigate()
    # Because 'if not ax' is True, should return False without calling get_navigate
    codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_with_truthy_object():
    """Test: ax_zoomable returns according to get_navigate even if object is always truthy."""
    class Truthy:
        def __bool__(self):
            return True
        def get_navigate(self):
            return False
    ax = Truthy()
    codeflash_output = Renderer.ax_zoomable(ax)

# 3. Large Scale Test Cases

def test_zoomable_large_batch_true():
    """Test: ax_zoomable returns True for a large batch of axes with get_navigate True."""
    axes = [DummyAx(True) for _ in range(500)]  # 500 is within the 1000 element limit
    for ax in axes:
        codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_large_batch_false():
    """Test: ax_zoomable returns False for a large batch of axes with get_navigate False."""
    axes = [DummyAx(False) for _ in range(500)]
    for ax in axes:
        codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_large_batch_mixed():
    """Test: ax_zoomable returns correct values for a large batch of mixed axes."""
    axes = [DummyAx(i % 2 == 0) for i in range(500)]  # Even indices True, odd False
    for i, ax in enumerate(axes):
        expected = (i % 2 == 0)
        codeflash_output = Renderer.ax_zoomable(ax)

def test_zoomable_large_batch_none_and_empty():
    """Test: ax_zoomable returns False for a large batch of None and empty objects."""
    batch = [None, 0, '', [], {}] * 100  # 500 elements, all falsy
    for obj in batch:
        codeflash_output = Renderer.ax_zoomable(obj)
# 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-Renderer.ax_zoomable-mb2iklvw and push.