@@ -15,7 +15,7 @@ async def test_icons_and_website_url():
15
15
test_icon = Icon (
16
16
src = "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNkYPhfDwAChwGA60e6kgAAAABJRU5ErkJggg==" ,
17
17
mimeType = "image/png" ,
18
- sizes = "1x1" ,
18
+ sizes = [ "1x1" ] ,
19
19
)
20
20
21
21
# Create server with website URL and icon
@@ -80,9 +80,9 @@ async def test_multiple_icons():
80
80
"""Test that multiple icons can be added to tools, resources, and prompts."""
81
81
82
82
# Create multiple test icons
83
- icon1 = Icon (src = "data:image/png;base64,icon1" , mimeType = "image/png" , sizes = "16x16" )
84
- icon2 = Icon (src = "data:image/png;base64,icon2" , mimeType = "image/png" , sizes = "32x32" )
85
- icon3 = Icon (src = "data:image/png;base64,icon3" , mimeType = "image/png" , sizes = "64x64" )
83
+ icon1 = Icon (src = "data:image/png;base64,icon1" , mimeType = "image/png" , sizes = [ "16x16" ] )
84
+ icon2 = Icon (src = "data:image/png;base64,icon2" , mimeType = "image/png" , sizes = [ "32x32" ] )
85
+ icon3 = Icon (src = "data:image/png;base64,icon3" , mimeType = "image/png" , sizes = [ "64x64" ] )
86
86
87
87
mcp = FastMCP ("MultiIconServer" )
88
88
@@ -98,9 +98,9 @@ def multi_icon_tool() -> str:
98
98
tool = tools [0 ]
99
99
assert tool .icons is not None
100
100
assert len (tool .icons ) == 3
101
- assert tool .icons [0 ].sizes == "16x16"
102
- assert tool .icons [1 ].sizes == "32x32"
103
- assert tool .icons [2 ].sizes == "64x64"
101
+ assert tool .icons [0 ].sizes == [ "16x16" ]
102
+ assert tool .icons [1 ].sizes == [ "32x32" ]
103
+ assert tool .icons [2 ].sizes == [ "64x64" ]
104
104
105
105
106
106
async def test_no_icons_or_website ():
0 commit comments