diff --git a/tests/conftest.py b/tests/conftest.py
index 5a197b9cd..e64d23e37 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -7,6 +7,7 @@
def water() -> ase.Atoms:
return ase.build.molecule("H2O")
+
@pytest.fixture
def ase_s22() -> list[ase.Atoms]:
- return list(ase.collections.s22)
\ No newline at end of file
+ return list(ase.collections.s22)
diff --git a/tests/test_zndraw.py b/tests/test_zndraw.py
index 9aaadec5c..28fe8d7fc 100644
--- a/tests/test_zndraw.py
+++ b/tests/test_zndraw.py
@@ -5,10 +5,10 @@
# response = client.app.get("/")
# assert b"
Hello, World!
" in response.data
+import numpy.testing as npt
import pytest
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
-import numpy.testing as npt
from zndraw import ZnDraw
@@ -24,6 +24,7 @@ def vis() -> ZnDraw:
yield visualizer
visualizer.close()
+
@pytest.mark.chrome
def test_gui_running(water):
vis = ZnDraw(token="test_token")
@@ -33,6 +34,7 @@ def test_gui_running(water):
assert "ZnDraw" in driver.title
vis.close()
+
@pytest.mark.chrome
def test_vis_atoms(vis, water):
vis[0] = water
@@ -41,12 +43,14 @@ def test_vis_atoms(vis, water):
assert vis.step == 0
+
@pytest.mark.chrome
def test_vis_selection(vis, water):
vis[0] = water
vis.selection = [1, 2]
assert vis.selection == [1, 2]
+
@pytest.mark.parametrize("display_new", [True, False])
@pytest.mark.chrome
def test_vis_step(vis, ase_s22, display_new):
@@ -61,6 +65,7 @@ def test_vis_step(vis, ase_s22, display_new):
vis.step = 10
assert vis.step == 10
+
@pytest.mark.chrome
def test_vis_points(vis, water):
vis[0] = water