File tree Expand file tree Collapse file tree 1 file changed +17
-4
lines changed Expand file tree Collapse file tree 1 file changed +17
-4
lines changed Original file line number Diff line number Diff line change @@ -484,10 +484,14 @@ def test_fit_bounds(self):
484
484
def test_image_overlay (self ):
485
485
"""Test image overlay"""
486
486
from numpy .random import random
487
- import imageio
487
+ from folium .utilities import write_png
488
+ import base64
488
489
489
490
data = random ((100 ,100 ))
490
- imageio .imwrite ('test.png' , data )
491
+ png_str = write_png (data )
492
+ with open ('data.png' , 'wb' ) as f :
493
+ f .write (png_str )
494
+ inline_image_url = "data:image/png;base64," + base64 .b64encode (png_str )
491
495
492
496
image_tpl = self .env .get_template ('image_layer.js' )
493
497
image_name = 'Image_Overlay'
@@ -502,8 +506,17 @@ def test_image_overlay(self):
502
506
'image_opacity' : image_opacity
503
507
})
504
508
505
- self .map .image_overlay (data )
509
+ self .map .image_overlay (data , filename = image_url )
510
+ assert image_rendered in self .map .template_vars ['image_layers' ]
511
+
506
512
513
+ image_rendered = image_tpl .render ({'image_name' : image_name ,
514
+ 'image_url' : inline_image_url ,
515
+ 'image_bounds' : image_bounds ,
516
+ 'image_opacity' : image_opacity
517
+ })
518
+
519
+ self .map .image_overlay (data )
507
520
assert image_rendered in self .map .template_vars ['image_layers' ]
508
-
521
+
509
522
You can’t perform that action at this time.
0 commit comments