update the test workflow with Element testing

osscar-org · Jun 8, 2024 · 57bf007 · 57bf007
1 parent bf5cc30
commit 57bf007
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/test/test.py b/test/test.py
@@ -23,6 +23,13 @@ def setup_method(self, method):
   def teardown_method(self, method):
     self.driver.quit()
 
+  def test_elements(self):
+    try:
+      self.find_element(By.CLASS_NAME, 'element-Si')
+      print("Element Si is fund!")
+    else:
+      raise Exception("Element Si is NOT found!!!")
+
   def download_widget_image(self):
     self.driver.get("http://localhost:8383/voila/render/example.ipynb")
     self.driver.set_window_size(1280, 1080)
@@ -37,5 +44,6 @@ def download_widget_image(self):
 test = test_widget()
 test.setup_method('Chrome')
 test.download_widget_image()
+test.test_elements()
 test.teardown_method('Chrome')