update the test workflow

.github/workflows/voila-test.yml → .github/workflows/screenshot-comparison.yml

@@ -1,4 +1,4 @@
-name: Voila test
+name: screenshot comparison
 
 on:
   push:
@@ -9,27 +9,27 @@ on:
       - '*'
 
 jobs:
-  voila:
+  check-screenshot:
+    name: Check Screenshot
     runs-on: ubuntu-latest
     steps:
     - name: Checkout
-      uses: actions/checkout@v2
+      uses: actions/checkout@v4
     - name: Install node
-      uses: actions/setup-node@v1
+      uses: actions/setup-node@v4
       with:
-       node-version: '18.x'
+       node-version: '21.x'
     - name: Install Python
-      uses: actions/setup-python@v2
+      uses: actions/setup-python@v5
       with:
-        python-version: '3.9'
+        python-version: '3.11'
         architecture: 'x64'
     - name: Install dependencies
       run: 
         pip install --upgrade pip
         pip install --upgrade jupyterlab
         pip install --upgrade voila
         pip install --upgrade voila-osscar-template
-        pip install --upgrade ase
         pip install -e .
 
     - name: Run the voila server
@@ -45,13 +45,5 @@ jobs:
         chromedriver --url-base=/wd/hub &
         sudo Xvfb -ac :99 -screen 0 1280x1024x24 > /dev/null 2>&1 & # optional
         python3 $GITHUB_WORKSPACE/test/test.py
-
-    - uses: actions/upload-artifact@v2
-      with:
-        name: quantum-screenshot
-        path: './*.png'
-
-    - name: Run the figures test
-      run: |
         python3 $GITHUB_WORKSPACE/test/test_figures.py
 

.github/workflows/widget-test.yml

@@ -0,0 +1,53 @@
+name: widget test
+
+on:
+  push:
+    branches:
+      - '*'
+  pull_request:
+    branches:
+      - '*'
+
+jobs:
+  check-elements:
+    name: Check Element
+    runs-on: ubuntu-latest
+    steps:
+    - name: Checkout
+      uses: actions/checkout@v4
+    - name: Install node
+      uses: actions/setup-node@v4
+      with:
+       node-version: '21.x'
+    - name: Install Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: '3.11'
+        architecture: 'x64'
+    - name: Install dependencies
+      run: 
+        pip install --upgrade pip
+        pip install --upgrade jupyterlab
+        pip install --upgrade voila
+        pip install --upgrade voila-osscar-template
+        pip install -e .
+
+    - name: Run the voila server
+      run: |
+        voila --template=osscar --enable_nbextensions=True example/ --port 8383 --no-browser &
+        pip install --upgrade pytest
+        pip install --upgrade selenium
+        pip install --upgrade Pillow
+
+    - uses: nanasess/setup-chromedriver@master
+    - run: |
+        export DISPLAY=:99
+        chromedriver --url-base=/wd/hub &
+        sudo Xvfb -ac :99 -screen 0 1280x1024x24 > /dev/null 2>&1 & # optional
+        python3 $GITHUB_WORKSPACE/test/test.py
+
+    - uses: actions/upload-artifact@v4
+      with:
+        name: artifact
+        path: './*.png'
+

test/test.py

@@ -23,6 +23,17 @@ def setup_method(self, method):
   def teardown_method(self, method):
     self.driver.quit()
 
+  def test_elements(self):
+    self.driver.get("http://localhost:8383/voila/render/example.ipynb")
+    self.driver.set_window_size(1280, 1080)
+    time.sleep(3)
+
+    try:
+      self.driver.find_element(By.CLASS_NAME, 'bz-canvas3d')
+      print("Brillouin zone visualizer is fund!")
+    except:
+      raise Exception("The visualizer is NOT found!!!")
+
   def download_widget_image(self):
     self.driver.get("http://localhost:8383/voila/render/example.ipynb")
     self.driver.set_window_size(1280, 720)
@@ -35,5 +46,6 @@ def download_widget_image(self):
 test = test_widget()
 test.setup_method('Chrome')
 test.download_widget_image()
+test.test_elements()
 test.teardown_method('Chrome')