|
16 | 16 |
|
17 | 17 | fig = {"data": [], "layout": {"title": {"text": "figure title"}}} |
18 | 18 |
|
| 19 | +def create_figure(width=None, height=None): |
| 20 | + """Create a simple figure with optional layout dimensions.""" |
| 21 | + layout = {} |
| 22 | + if width: |
| 23 | + layout["width"] = width |
| 24 | + if height: |
| 25 | + layout["height"] = height |
| 26 | + |
| 27 | + return go.Figure(data=[go.Scatter(x=[1, 2, 3], y=[1, 2, 3])], layout=layout) |
| 28 | + |
| 29 | +def parse_svg_dimensions(svg_bytes): |
| 30 | + """Parse width and height from SVG bytes.""" |
| 31 | + svg_str = svg_bytes.decode("utf-8") |
| 32 | + root = ET.fromstring(svg_str) |
| 33 | + width = root.get("width") |
| 34 | + height = root.get("height") |
| 35 | + return int(width) if width else None, int(height) if height else None |
| 36 | + |
19 | 37 |
|
20 | 38 | def check_image(path_or_buffer, size=(700, 500), format="PNG"): |
21 | 39 | if format == "PDF": |
@@ -317,26 +335,6 @@ def test_get_chrome(): |
317 | 335 | mock_get_chrome.assert_called_once() |
318 | 336 |
|
319 | 337 |
|
320 | | -def create_figure(width=None, height=None): |
321 | | - """Create a simple figure with optional layout dimensions.""" |
322 | | - layout = {} |
323 | | - if width: |
324 | | - layout["width"] = width |
325 | | - if height: |
326 | | - layout["height"] = height |
327 | | - |
328 | | - return go.Figure(data=[go.Scatter(x=[1, 2, 3], y=[1, 2, 3])], layout=layout) |
329 | | - |
330 | | - |
331 | | -def parse_svg_dimensions(svg_bytes): |
332 | | - """Parse width and height from SVG bytes.""" |
333 | | - svg_str = svg_bytes.decode("utf-8") |
334 | | - root = ET.fromstring(svg_str) |
335 | | - width = root.get("width") |
336 | | - height = root.get("height") |
337 | | - return int(width) if width else None, int(height) if height else None |
338 | | - |
339 | | - |
340 | 338 | def test_width_height_priority(): |
341 | 339 | """Test width/height priority: arguments > layout.width/height > defaults.""" |
342 | 340 |
|
|
0 commit comments