Skip to content

Commit 1d3c144

Browse files
committed
fieldline test with seed, new symmetry baseline from bounding box fix
1 parent f4bb518 commit 1d3c144

4 files changed

Lines changed: 11 additions & 6 deletions

File tree

ngsolve_webgpu/cf.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1076,6 +1076,7 @@ def __init__(
10761076
direction: int = 0,
10771077
colormap=None,
10781078
clipping=None,
1079+
seed: int | None = None,
10791080
):
10801081
import ngsolve as ngs
10811082
self.fieldline_options = {
@@ -1086,6 +1087,7 @@ def __init__(
10861087
"tolerance": tolerance,
10871088
"direction": direction,
10881089
}
1090+
self.seed = seed
10891091
self.cf = cf
10901092
if isinstance(start_region, ngs.Mesh):
10911093
self.mesh = start_region
@@ -1107,8 +1109,11 @@ def get_bounding_box(self):
11071109
return ([pmin[0], pmin[1], pmin[2]], [pmax[0], pmax[1], pmax[2]])
11081110

11091111
def update(self, options):
1112+
import numpy as np
11101113
from ngsolve.webgui import FieldLines
11111114

1115+
if self.seed is not None:
1116+
np.random.seed(self.seed)
11121117
data = FieldLines(self.cf, self.start_region, **self.fieldline_options)
11131118
bbox = self.mesh.ngmesh.bounding_box
11141119
thickness = (bbox[1] - bbox[0]).Norm() * self.fieldline_options["thickness"]

tests/baselines/fieldlines.png

Lines changed: 2 additions & 2 deletions
Loading
Lines changed: 2 additions & 2 deletions
Loading

tests/test_vectors.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def test_fieldlines(self, webgpu_env):
101101
webgpu_env.ensure_canvas(600, 600)
102102
mesh = ngs.Mesh(ngs.unit_cube.GenerateMesh(maxh=0.5))
103103
cf = ngs.CF((ngs.x, ngs.y, ngs.z))
104-
renderer = FieldLines(cf, mesh, num_lines=20)
104+
renderer = FieldLines(cf, mesh, num_lines=20, seed=42)
105105
scene = wj.Draw([renderer], 600, 600)
106106

107-
webgpu_env.assert_matches_baseline(scene, "fieldlines.png", threshold=0.05)
107+
webgpu_env.assert_matches_baseline(scene, "fieldlines.png")

0 commit comments

Comments
 (0)