bokeh__bokeh-13289

{
  "status": "ok",
  "stdout": "src/bokeh/models/renderers/glyph_renderer.py:170:    def construct_color_bar(self, **kwargs: Any) -> ColorBar:\nsrc/bokeh/models/renderers/glyph_renderer.py:223:            raise ValueError(f\"construct_color_bar does not handle glyph type {type(self.glyph).__name__}\")\nsrc/bokeh/models/renderers/contour_renderer.py:100:    def construct_color_bar(self, **kwargs) -> ContourColorBar:\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "filepath": "src/bokeh/models/renderers/glyph_renderer.py",
  "content": "    def construct_color_bar(self, **kwargs: Any) -> ColorBar:\n        ''' Construct and return a new ``ColorBar`` for this ``GlyphRenderer``.\n\n        The function will check for a color mapper on an appropriate property\n        of the GlyphRenderer's main glyph, in this order:\n\n        * ``fill_color.transform`` for FillGlyph\n        * ``line_color.transform`` for LineGlyph\n        * ``text_color.transform`` for TextGlyph\n        * ``color_mapper`` for Image\n\n        In general, the function will \"do the right thing\" based on glyph type.\n        If different behavior is needed, ColorBars can be constructed by hand.\n\n        Extra keyword arguments may be passed in to control ``ColorBar``\n        properties such as `title`.\n\n        Returns:\n            ColorBar\n\n        '''\n        from ...core.property.vectorization import Field\n        from ..annotations import ColorBar\n        from ..glyphs import (\n            FillGlyph,\n            Image,\n            LineGlyph,\n            TextGlyph,\n        )\n        from ..mappers import ColorMapper\n\n        if isinstance(self.glyph, FillGlyph):\n            fill_color = self.glyph.fill_color\n            if not (isinstance(fill_color, Field) and isinstance(fill_color.transform, ColorMapper)):\n                raise ValueError(\"expected fill_color to be a field with a ColorMapper transform\")\n            return ColorBar(color_mapper=fill_color.transform, **kwargs)\n\n        elif isinstance(self.glyph, LineGlyph):\n            line_color = self.glyph.line_color\n            if not (isinstance(line_color, Field) and isinstance(line_color.transform, ColorMapper)):\n                raise ValueError(\"expected line_color to be a field with a ColorMapper transform\")\n            return ColorBar(color_mapper=line_color.transform, **kwargs)\n\n        elif isinstance(self.glyph, TextGlyph):\n            text_color = self.glyph.text_color\n            if not (isinstance(text_color, Field) and isinstance(text_color.transform, ColorMapper)):\n                raise ValueError(\"expected text_color to be a field with a ColorMapper transform\")\n            return ColorBar(color_mapper=text_color.transform, **kwargs)\n\n        elif type(self.glyph) is Image:\n            return ColorBar(color_mapper=self.glyph.color_mapper, **kwargs)\n\n        else:\n            raise ValueError(f\"construct_color_bar does not handle glyph type {type(self.glyph).__name__}\")\n\n#-----------------------------------------------------------------------------\n# Dev API\n#-----------------------------------------------------------------------------\n\n#-----------------------------------------------------------------------------\n# Private API",
  "start_line": 170,
  "end_line": 230,
  "total_lines": 235,
  "is_truncated": false
}

{
  "status": "ok",
  "stdout": "src/bokeh/models/glyphs.py:834:class ImageStack(ImageBase):\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "filepath": "src/bokeh/models/glyphs.py",
  "content": "    image = NumberSpec(default=field(\"image\"), help=\"\"\"\n    The arrays of RGBA data for the images.\n    \"\"\")\n\nclass ImageStack(ImageBase):\n    ''' Render images given as 3D stacked arrays by flattening each stack into\n    an RGBA image using a ``StackColorMapper``.\n\n    The 3D arrays have shape (ny, nx, nstack) where ``nstack` is the number of\n    stacks. The ``color_mapper`` produces an RGBA value for each of the\n    (ny, nx) pixels by combining array values in the ``nstack`` direction.\n\n    '''\n\n    # explicit __init__ to support Init signatures\n    def __init__(self, *args, **kwargs) -> None:\n        super().__init__(*args, **kwargs)\n\n    _args = ('image', 'x', 'y', 'dw', 'dh', 'dilate')\n\n    image = NumberSpec(default=field(\"image\"), help=\"\"\"\n    The 3D arrays of data for the images.\n    \"\"\")\n\n    color_mapper = Instance(StackColorMapper, help=\"\"\"\n    ``ScalarColorMapper`` used to map the scalar data from ``image``\n    into RGBA values for display.\n\n    .. note::\n        The color mapping step happens on the client.\n    \"\"\")",
  "start_line": 830,
  "end_line": 860,
  "total_lines": 1835,
  "is_truncated": false
}

{
  "status": "ok",
  "stdout": "src/bokeh/models/glyphs.py:768:class Image(ImageBase):\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "filepath": "src/bokeh/models/glyphs.py",
  "content": "    origin = Enum(ImageOrigin, default=\"bottom_left\", help=\"\"\"\n    Defines the coordinate space of an image.\n    \"\"\")\n\n    anchor = Anchor(default=\"bottom_left\", help=\"\"\"\n    Position of the image should be anchored at the `x`, `y` coordinates.\n    \"\"\")\n\nclass Image(ImageBase):\n    ''' Render images given as scalar data together with a color mapper.\n\n    In addition to the defined model properties, ``Image`` also can accept\n    a keyword argument ``palette`` in place of an explicit ``color_mapper``.\n    The value should be a list of colors, or the name of one of the built-in\n    palettes in ``bokeh.palettes``. This palette will be used to automatically\n    construct a ``ColorMapper`` model for the ``color_mapper`` property.\n\n    If both ``palette`` and ``color_mapper`` are passed, a ``ValueError``\n    exception will be raised. If neither is passed, then the ``Greys9``\n    palette will be used as a default.\n",
  "start_line": 760,
  "end_line": 780,
  "total_lines": 1835,
  "is_truncated": false
}

{
  "status": "ok",
  "stdout": "718:class ImageBase(XYGlyph):\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "filepath": "src/bokeh/models/glyphs.py",
  "content": "class ImageBase(XYGlyph):\n\n    # explicit __init__ to support Init signatures\n    def __init__(self, *args, **kwargs) -> None:\n        super().__init__(*args, **kwargs)\n\n    x = NumberSpec(default=field(\"x\"), help=\"\"\"\n    The x-coordinates to locate the image anchors.\n    \"\"\")\n\n    y = NumberSpec(default=field(\"y\"), help=\"\"\"\n    The y-coordinates to locate the image anchors.\n    \"\"\")\n\n    dw = DistanceSpec(default=field(\"dw\"), help=\"\"\"\n    The widths of the plot regions that the images will occupy.\n\n    .. note::\n        This is not the number of pixels that an image is wide.\n        That number is fixed by the image itself.\n    \"\"\")\n\n    dh = DistanceSpec(default=field(\"dh\"), help=\"\"\"\n    The height of the plot region that the image will occupy.\n\n    .. note::\n        This is not the number of pixels that an image is tall.\n        That number is fixed by the image itself.\n    \"\"\")\n\n    image_props = Include(ImageProps, help=\"\"\"\n    The {prop} values for the images.\n    \"\"\")\n\n    dilate = Bool(False, help=\"\"\"\n    Whether to always round fractional pixel locations in such a way\n    as to make the images bigger.\n\n    This setting may be useful if pixel rounding errors are causing\n    images to have a gap between them, when they should appear flush.\n    \"\"\")\n\n    origin = Enum(ImageOrigin, default=\"bottom_left\", help=\"\"\"",
  "start_line": 718,
  "end_line": 760,
  "total_lines": 1835,
  "is_truncated": false
}

{
  "status": "ok",
  "filepath": "src/bokeh/models/glyphs.py",
  "content": "\n    In addition to the defined model properties, ``Image`` also can accept\n    a keyword argument ``palette`` in place of an explicit ``color_mapper``.\n    The value should be a list of colors, or the name of one of the built-in\n    palettes in ``bokeh.palettes``. This palette will be used to automatically\n    construct a ``ColorMapper`` model for the ``color_mapper`` property.\n\n    If both ``palette`` and ``color_mapper`` are passed, a ``ValueError``\n    exception will be raised. If neither is passed, then the ``Greys9``\n    palette will be used as a default.\n\n    '''\n\n    def __init__(self, *args, **kwargs) -> None:\n        if 'palette' in kwargs and 'color_mapper' in kwargs:\n            raise ValueError(\"only one of 'palette' and 'color_mapper' may be specified\")\n        elif 'color_mapper' not in kwargs:\n            # Use a palette (given or default)\n            palette = kwargs.pop('palette', 'Greys9')\n            mapper = LinearColorMapper(palette)\n            kwargs['color_mapper'] = mapper\n\n        super().__init__(*args, **kwargs)\n\n    _args = ('image', 'x', 'y', 'dw', 'dh', 'dilate')\n\n    # a hook to specify any additional kwargs handled by an initializer\n    _extra_kws = {\n        'palette': (\n            'str or list[color value]',\n            'a palette to construct a value for the color mapper property from',\n        ),\n    }\n\n    image = NumberSpec(default=field(\"image\"), help=\"\"\"\n    The arrays of scalar data for the images to be colormapped.\n    \"\"\")\n\n    color_mapper = Instance(ColorMapper, default=InstanceDefault(LinearColorMapper, palette=\"Greys9\"), help=\"\"\"\n    A ``ColorMapper`` to use to map the scalar data from ``image``\n    into RGBA values for display.\n\n    The name of a palette from ``bokeh.palettes`` may also be set, in which\n    case a ``LinearColorMapper`` configured with the named palette wil be used.\n\n    .. note::\n        The color mapping step happens on the client.\n    \"\"\").accepts(Enum(Palette), lambda pal: LinearColorMapper(palette=pal))\n\nclass ImageRGBA(ImageBase):\n    ''' Render images given as RGBA data.\n\n    '''\n\n    # explicit __init__ to support Init signatures\n    def __init__(self, *args, **kwargs) -> None:\n        super().__init__(*args, **kwargs)\n\n    _args = ('image', 'x', 'y', 'dw', 'dh', 'dilate')\n\n    image = NumberSpec(default=field(\"image\"), help=\"\"\"",
  "start_line": 770,
  "end_line": 830,
  "total_lines": 1835,
  "is_truncated": false
}

{
  "status": "ok",
  "filepath": "src/bokeh/models/glyphs.py",
  "content": "    ''' Render images given as RGBA data.\n\n    '''\n\n    # explicit __init__ to support Init signatures\n    def __init__(self, *args, **kwargs) -> None:\n        super().__init__(*args, **kwargs)\n\n    _args = ('image', 'x', 'y', 'dw', 'dh', 'dilate')\n\n    image = NumberSpec(default=field(\"image\"), help=\"\"\"\n    The arrays of RGBA data for the images.\n    \"\"\")\n\nclass ImageStack(ImageBase):\n    ''' Render images given as 3D stacked arrays by flattening each stack into",
  "start_line": 820,
  "end_line": 835,
  "total_lines": 1835,
  "is_truncated": false
}

{
  "status": "ok",
  "stdout": "Caught expected error: construct_color_bar does not handle glyph type ImageStack\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "filepath": "src/bokeh/models/renderers/glyph_renderer.py",
  "occurrences": 1,
  "strategy": "unescaped",
  "diff": "--- a/src/bokeh/models/renderers/glyph_renderer.py\n+++ b/src/bokeh/models/renderers/glyph_renderer.py\n@@ -193,6 +193,7 @@\n         from ..glyphs import (\n             FillGlyph,\n             Image,\n+            ImageStack,\n             LineGlyph,\n             TextGlyph,\n         )\n",
  "is_truncated": false
}

{
  "status": "ok",
  "filepath": "src/bokeh/models/renderers/glyph_renderer.py",
  "occurrences": 1,
  "strategy": "unescaped",
  "diff": "--- a/src/bokeh/models/renderers/glyph_renderer.py\n+++ b/src/bokeh/models/renderers/glyph_renderer.py\n@@ -217,7 +217,7 @@\n                 raise ValueError(\"expected text_color to be a field with a ColorMapper transform\")\n             return ColorBar(color_mapper=text_color.transform, **kwargs)\n \n-        elif type(self.glyph) is Image:\n+        elif isinstance(self.glyph, (Image, ImageStack)):\n             return ColorBar(color_mapper=self.glyph.color_mapper, **kwargs)\n \n         else:\n",
  "is_truncated": false
}

{
  "status": "ok",
  "stdout": "Success!\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "stdout": "Success? (Should have failed)\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "stdout": "Glyph type: <class 'bokeh.models.glyphs.Image'>\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "stdout": "ImageRGBA is Image: False\nImageRGBA is ImageStack: False\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "stdout": "718:class ImageBase(XYGlyph):\n768:class Image(ImageBase):\n819:class ImageRGBA(ImageBase):\n834:class ImageStack(ImageBase):\n862:class ImageURL(XYGlyph):\n",
  "stderr": "",
  "exit_code": 0
}

{
  "status": "ok",
  "filepath": "src/bokeh/models/glyphs.py",
  "content": "    \"\"\")\n\nclass ImageURL(XYGlyph):\n    ''' Render images loaded from given URLs.\n\n    '''\n\n    # explicit __init__ to support Init signatures\n    def __init__(self, *args, **kwargs) -> None:\n        super().__init__(*args, **kwargs)\n\n    __example__ = \"examples/reference/models/ImageURL.py\"\n\n    _args = ('url', 'x', 'y', 'w', 'h', 'angle', 'dilate')\n\n    url = StringSpec(default=field(\"url\"), help=\"\"\"\n    The URLs to retrieve images from.\n\n    .. note::\n        The actual retrieving and loading of the images happens on\n        the client.\n    \"\"\")\n\n    x = NumberSpec(default=field(\"x\"), help=\"\"\"\n    The x-coordinates to locate the image anchors.\n    \"\"\")\n\n    y = NumberSpec(default=field(\"y\"), help=\"\"\"\n    The y-coordinates to locate the image anchors.\n    \"\"\")\n\n    w = NullDistanceSpec(help=\"\"\"\n    The width of the plot region that the image will occupy in data space.\n\n    The default value is ``None``, in which case the image will be displayed\n    at its actual image size (regardless of the units specified here).\n    \"\"\")\n\n    h = NullDistanceSpec(help=\"\"\"\n    The height of the plot region that the image will occupy in data space.\n\n    The default value is ``None``, in which case the image will be displayed\n    at its actual image size (regardless of the units specified here).\n    \"\"\")\n\n    angle = AngleSpec(default=0, help=\"\"\"\n    The angles to rotate the images, as measured from the horizontal.\n    \"\"\")\n\n    global_alpha = NumberSpec(1.0, help=\"\"\"\n    An overall opacity that each image is rendered with (in addition\n    to any inherent alpha values in the image itself).\n    \"\"\")\n\n    dilate = Bool(False, help=\"\"\"\n    Whether to always round fractional pixel locations in such a way\n    as to make the images bigger.\n\n    This setting may be useful if pixel rounding errors are causing\n    images to have a gap between them, when they should appear flush.\n    \"\"\")\n\n    anchor = Anchor(default=\"top_left\", help=\"\"\"\n    Position of the image should be anchored at the `x`, `y` coordinates.\n    \"\"\")\n\n    retry_attempts = Int(0, help=\"\"\"\n    Number of attempts to retry loading the images from the specified URL.\n    Default is zero.\n    \"\"\")\n",
  "start_line": 860,
  "end_line": 930,
  "total_lines": 1835,
  "is_truncated": false
}

{
  "status": "ok",
  "stdout": "tests/unit/bokeh/plotting/test_contour.py:269:    color_bar = cr.construct_color_bar()\ntests/unit/bokeh/models/test_glyph_renderer.py:49:class TestGlyphRenderer_construct_color_bar:\ntests/unit/bokeh/models/test_glyph_renderer.py:55:        cb = renderer.construct_color_bar(title=\"Title\")\ntests/unit/bokeh/models/test_glyph_renderer.py:66:            renderer.construct_color_bar()\ntests/unit/bokeh/models/test_glyph_renderer.py:72:        cb = renderer.construct_color_bar(title=\"Title\")\ntests/unit/bokeh/models/test_glyph_renderer.py:83:            renderer.construct_color_bar()\ntests/unit/bokeh/models/test_glyph_renderer.py:89:        cb = renderer.construct_color_bar(title=\"Title\")\ntests/unit/bokeh/models/test_glyph_renderer.py:100:            renderer.construct_color_bar()\ntests/unit/bokeh/models/test_glyph_renderer.py:106:        cb = renderer.construct_color_bar(title=\"Title\")\ntests/unit/bokeh/models/test_glyph_renderer.py:115:        msg = f\"construct_color_bar does not handle glyph type {type(renderer.glyph).__name__}\"\ntests/unit/bokeh/models/test_glyph_renderer.py:117:            renderer.construct_color_bar()\n",
  "stderr": "",
  "exit_code": 0,
  "budget_warning": "Only 10 tool call(s) remaining (20/30 used). Finalize your edits and call submit_patch soon."
}

{
  "status": "ok",
  "filepath": "tests/unit/bokeh/models/test_glyph_renderer.py",
  "content": "#-----------------------------------------------------------------------------\n# Setup\n#-----------------------------------------------------------------------------\n\n#-----------------------------------------------------------------------------\n# General API\n#-----------------------------------------------------------------------------\n\n\nclass TestGlyphRenderer_construct_color_bar:\n\n    @pytest.mark.parametrize(\"mapper\", (linear_cmap, log_cmap))\n    def test_fill_good(self, mapper):\n        renderer = bmr.GlyphRenderer(data_source=ColumnDataSource())\n        renderer.glyph = Circle(fill_color=linear_cmap(\"foo\", \"Viridis256\", 0, 100))\n        cb = renderer.construct_color_bar(title=\"Title\")\n        assert isinstance(cb, ColorBar)\n        assert cb.color_mapper is renderer.glyph.fill_color.transform\n        assert cb.title == \"Title\"\n\n    def test_fill_missing_transform(self):\n        renderer = bmr.GlyphRenderer(data_source=ColumnDataSource())\n        renderer.glyph = Circle()\n\n        msg = \"expected fill_color to be a field with a ColorMapper transform\"\n        with pytest.raises(ValueError, match=msg):\n            renderer.construct_color_bar()\n\n    @pytest.mark.parametrize(\"mapper\", (linear_cmap, log_cmap))\n    def test_line_good(self, mapper):\n        renderer = bmr.GlyphRenderer(data_source=ColumnDataSource())\n        renderer.glyph = MultiLine(line_color=linear_cmap(\"foo\", \"Viridis256\", 0, 100))\n        cb = renderer.construct_color_bar(title=\"Title\")\n        assert isinstance(cb, ColorBar)\n        assert cb.color_mapper is renderer.glyph.line_color.transform\n        assert cb.title == \"Title\"\n\n    def test_line_missing_transform(self):\n        renderer = bmr.GlyphRenderer(data_source=ColumnDataSource())\n        renderer.glyph = MultiLine()\n\n        msg = \"expected line_color to be a field with a ColorMapper transform\"\n        with pytest.raises(ValueError, match=msg):\n            renderer.construct_color_bar()\n\n    @pytest.mark.parametrize(\"mapper\", (linear_cmap, log_cmap))\n    def test_text_good(self, mapper):\n        renderer = bmr.GlyphRenderer(data_source=ColumnDataSource())\n        renderer.glyph = Text(text_color=linear_cmap(\"foo\", \"Viridis256\", 0, 100))\n        cb = renderer.construct_color_bar(title=\"Title\")\n        assert isinstance(cb, ColorBar)\n        assert cb.color_mapper is renderer.glyph.text_color.transform\n        assert cb.title == \"Title\"\n\n    def test_text_missing_transform(self):\n        renderer = bmr.GlyphRenderer(data_source=ColumnDataSource())\n        renderer.glyph = Text()\n\n        msg = \"expected text_color to be a field with a ColorMapper transform\"\n        with pytest.raises(ValueError, match=msg):\n            renderer.construct_color_bar()\n\n    @pytest.mark.parametrize(\"mapper\", (linear_cmap, log_cmap))\n    def test_image_good(self, mapper):\n        renderer = bmr.GlyphRenderer(data_source=ColumnDataSource())\n        renderer.glyph = Image(color_mapper=linear_cmap(\"foo\", \"Viridis256\", 0, 100).transform)\n        cb = renderer.construct_color_bar(title=\"Title\")\n        assert isinstance(cb, ColorBar)\n        assert cb.color_mapper is renderer.glyph.color_mapper\n        assert cb.title == \"Title\"\n\n    def test_unknown_glyph_type(self):\n        renderer = bmr.GlyphRenderer(data_source=ColumnDataSource())\n        renderer.glyph = ImageRGBA()\n\n        msg = f\"construct_color_bar does not handle glyph type {type(renderer.glyph).__name__}\"\n        with pytest.raises(ValueError, match=msg):\n            renderer.construct_color_bar()\n\nclass TestGlyphRenderer_check_bad_column_name:\n",
  "start_line": 40,
  "end_line": 120,
  "total_lines": 212,
  "is_truncated": false,
  "budget_warning": "Only 9 tool call(s) remaining (21/30 used). Finalize your edits and call submit_patch soon."
}

{
  "status": "ok",
  "stdout": ".....................                                                    [100%]\n21 passed in 0.17s\n",
  "stderr": "/opt/miniconda3/envs/testbed/lib/python3.10/site-packages/pytest_asyncio/plugin.py:208: PytestDeprecationWarning: The configuration option \"asyncio_default_fixture_loop_scope\" is unset.\nThe event loop scope for asynchronous fixtures will default to the fixture caching scope. Future versions of pytest-asyncio will default the loop scope for asynchronous fixtures to function scope. Set the default fixture loop scope explicitly in order to avoid unexpected behavior in the future. Valid fixture loop scopes are: \"function\", \"class\", \"module\", \"package\", \"session\"\n\n  warnings.warn(PytestDeprecationWarning(_DEFAULT_FIXTURE_LOOP_SCOPE_UNSET))\n",
  "exit_code": 0,
  "budget_warning": "Only 8 tool call(s) remaining (22/30 used). Finalize your edits and call submit_patch soon."
}

{
  "status": "ok",
  "stdout": "Success!\n",
  "stderr": "",
  "exit_code": 0,
  "budget_warning": "Only 7 tool call(s) remaining (23/30 used). Finalize your edits and call submit_patch soon."
}

{
  "status": "ok",
  "stdout": " M src/bokeh/models/renderers/glyph_renderer.py\ndiff --git a/src/bokeh/models/renderers/glyph_renderer.py b/src/bokeh/models/renderers/glyph_renderer.py\nindex 4172018..1da7aa8 100644\n--- a/src/bokeh/models/renderers/glyph_renderer.py\n+++ b/src/bokeh/models/renderers/glyph_renderer.py\n@@ -193,6 +193,7 @@ class GlyphRenderer(DataRenderer):\n         from ..glyphs import (\n             FillGlyph,\n             Image,\n+            ImageStack,\n             LineGlyph,\n             TextGlyph,\n         )\n@@ -216,7 +217,7 @@ class GlyphRenderer(DataRenderer):\n                 raise ValueError(\"expected text_color to be a field with a ColorMapper transform\")\n             return ColorBar(color_mapper=text_color.transform, **kwargs)\n \n-        elif type(self.glyph) is Image:\n+        elif isinstance(self.glyph, (Image, ImageStack)):\n             return ColorBar(color_mapper=self.glyph.color_mapper, **kwargs)\n \n         else:\n",
  "stderr": "",
  "exit_code": 0,
  "budget_warning": "Only 6 tool call(s) remaining (24/30 used). Finalize your edits and call submit_patch soon."
}

{
  "status": "ok",
  "patch_size": 877,
  "files_changed": 1
}