Quit driver when done. Add test

jwql/tests/test_utils.py

@@ -22,7 +22,11 @@
 from pathlib import Path
 import pytest
 
-from jwql.utils.utils import copy_files, get_config, filename_parser, filesystem_path, _validate_config
+from bokeh.models import LinearColorMapper
+from bokeh.plotting import figure
+import numpy as np
+
+from jwql.utils.utils import copy_files, get_config, filename_parser, filesystem_path, save_png, _validate_config
 
 
 # Determine if tests are being run on Github Actions
@@ -479,6 +483,19 @@ def test_filesystem_path():
     assert check == location
 
 
+def test_save_png():
+    """Test that we can create a png file"""
+    plot = figure(title='test',tools='')
+    image = np.zeros((200,200))
+    image[100:105, 100:105] = 1
+    ny, nx = image.shape
+    mapper = LinearColorMapper(palette='Viridis256', low=0 ,high=1.1)
+    imgplot = plot.image(image=[image], x=0, y=0, dw=nx, dh=ny, color_mapper=mapper, level="image")
+    #save_png(plot, filename='test.png')
+    from bokeh.io import export_png
+    export_png(plot, filename='test.png')
+
+
 @pytest.mark.skipif(ON_GITHUB_ACTIONS, reason='Requires access to central storage.')
 def test_validate_config():
     """Test that the config validator works."""

jwql/utils/utils.py

@@ -775,6 +775,7 @@ def save_png(fig, filename=''):
     options.add_argument('-headless')
     driver = webdriver.Firefox(options=options)
     export_png(fig, filename=filename, webdriver=driver)
+    driver.quit()
 
 
 def grouper(iterable, chunksize):