update the test by clicking the matplotlib download button

test/test.py

@@ -2,7 +2,9 @@
 import pytest
 import time
 import json
+import os
 from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.common.by import By
 from selenium.webdriver.common.action_chains import ActionChains
 from selenium.webdriver.support import expected_conditions
@@ -12,9 +14,12 @@
 
 class TestTest01():
   def setup_method(self, method):
-    self.driver = webdriver.Chrome()
+    options = Options()
+    cwd = os.getcwd()
+    options.add_experimental_option("prefs", {"download.default_directory": cwd})
+    self.driver = webdriver.Chrome(options=options)
     self.vars = {}
-  
+
   def teardown_method(self, method):
     self.driver.quit()
 
@@ -34,8 +39,13 @@ def test_asymmetricwell(self):
     time.sleep(5)
     self.driver.execute_script("window.scrollTo(0, 400)")
     self.driver.find_element(By.CSS_SELECTOR, "label:nth-child(2) > input").click()
-    time.sleep(10)
-    self.driver.save_screenshot("asymmetricwell.png")
+    time.sleep(5)
+
+    download = self.driver.find_elements(By.XPATH, "//button[@title='Download plot']")[0]
+    actions = ActionChains(self.driver)
+    actions.move_to_element(download).click().perform()
+    time.sleep(3)
+    self.driver.save_screenshot("test.png")
 
 test = TestTest01()
 test.setup_method('Chrome')

test/test_figures.py

@@ -1,6 +1,6 @@
 from PIL import Image, ImageChops, ImageStat
 
-image1 = Image.open('asymmetricwell.png') 
+image1 = Image.open('Figure 1.png') 
 image2 = Image.open('test/asymmetricwell.png') 
 
 diff = ImageChops.difference(image1, image2)