From cd482a0142ff3c7604291f7354fa31d0c63f06ca Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 9 Jul 2015 18:34:25 -0500
Subject: [PATCH] Add vis option to box test
---
test/test_meshmode.py | 18 ++++++++++++++++--
1 file changed, 16 insertions(+), 2 deletions(-)
diff --git a/test/test_meshmode.py b/test/test_meshmode.py
index 086653ef..7c6e325b 100644
--- a/test/test_meshmode.py
+++ b/test/test_meshmode.py
@@ -382,9 +382,23 @@ def test_rect_mesh(do_plot=False):
pt.show()
-def test_box_mesh():
+def test_box_mesh(ctx_getter, visualize=False):
from meshmode.mesh.generation import generate_box_mesh
- generate_box_mesh(3*(np.linspace(0, 1, 5),))
+ mesh = generate_box_mesh(3*(np.linspace(0, 1, 5),))
+
+ if visualize:
+ from meshmode.discretization import Discretization
+ from meshmode.discretization.poly_element import \
+ PolynomialWarpAndBlendGroupFactory
+ cl_ctx = ctx_getter()
+ queue = cl.CommandQueue(cl_ctx)
+
+ discr = Discretization(cl_ctx, mesh,
+ PolynomialWarpAndBlendGroupFactory(1))
+
+ from meshmode.discretization.visualization import make_visualizer
+ vis = make_visualizer(queue, discr, 1)
+ vis.write_vtk_file("box.vtu", [])
def test_as_python():
--
GitLab