grid = vtki.UnstructuredGrid(filename)
for i in range(result.nsets):
assert 'nodal_solution%03d' % i in grid.point_arrays
arr = grid.point_arrays['nodal_solution%03d' % i]
assert np.allclose(arr, result.nodal_solution(i)[1], atol=1E-5)
assert 'nodal_stress%03d' % i in grid.point_arrays
arr = grid.point_arrays['nodal_stress%03d' % i]
assert np.allclose(arr,
result.nodal_stress(i)[1],
atol=1E-5,
equal_nan=True)
@pytest.mark.skipif(not running_xserver(), reason="Requires active X Server")
def test_plot_component():
"""
# create example file for component plotting
ansys = pyansys.ANSYS('/usr/ansys_inc/v182/ansys/bin/ansys182')
ansys.Cdread('db', examples.hexarchivefile)
# ansys.open_gui()
ansys.Esel('S', 'ELEM', vmin=1, vmax=20)
ansys.Nsle('S', 1)
ansys.Cm('MY_COMPONENT', 'NODE')
ansys.Nsel('S', 'NODE', '', 1, 40)
ansys.Cm('MY_OTHER_COMPONENT', 'NODE')
ansys.Allsel()
import os
import sys
from subprocess import PIPE, Popen
from weakref import proxy
import imageio
import numpy as np
import pytest
import vtki
from vtki import examples
from vtki.plotting import running_xserver
NO_PLOTTING = not running_xserver()
try:
if os.environ['ALLOW_PLOTTING'] == 'True':
NO_PLOTTING = False
except KeyError:
pass
if __name__ != '__main__':
OFF_SCREEN = 'pytest' in sys.modules
else:
OFF_SCREEN = False
sphere = vtki.Sphere()
sphere_b = vtki.Sphere(1.0)
sphere_c = vtki.Sphere(2.0)
@pytest.mark.skipif(NO_PLOTTING, reason="Requires X11")
import numpy as np
import pytest
import vtk
import vtki
from vtki import examples as ex
from vtki.plotting import running_xserver
if __name__ != '__main__':
OFF_SCREEN = 'pytest' in sys.modules
else:
OFF_SCREEN = False
@pytest.mark.skipif(not running_xserver(), reason="Requires X11")
def test_export_single(tmpdir):
filename = str(tmpdir.mkdir("tmpdir").join('scene'))
data = ex.load_airplane()
# Create the scene
plotter = vtki.Plotter(off_screen=OFF_SCREEN)
plotter.add_mesh(data)
plotter.export_vtkjs(filename)
cpos_out = plotter.show() # Export must be called before showing!
plotter.close()
# Now make sure the file is there
assert os.path.isfile('{}.vtkjs'.format(filename))
@pytest.mark.skipif(not running_xserver(), reason="Requires X11")
def test_export_multi(tmpdir):
class TestCyclicResultReader(object):
# avoids errors in collection
result_file = os.path.join(data_path, 'cyclic_%s.rst' % rver)
try:
# test if raw results are being read properly by using the normal result reader
result = pyansys.Result(result_file)
if rver == 'v182':
ansys = pyansys.ANSYS('/usr/ansys_inc/v182/ansys/bin/ansys182',
override=True, jobname=rver, loglevel='DEBUG',
interactive_plotting=False, prefer_pexpect=True)
else:
ansys = pyansys.ANSYS('/usr/ansys_inc/v150/ansys/bin/ansys150',
override=True, jobname=rver, loglevel='DEBUG',
interactive_plotting=False, prefer_pexpect=True)
# copy result file to ansys's temporary path
copyfile(result_file, os.path.join(ansys.path, '%s.rst' % rver))
# setup ansys for output without line breaks
ansys.Post1()
ansys.Set(1, 1)
ansys.Header('OFF', 'OFF', 'OFF', 'OFF', 'OFF', 'OFF')
nsigfig = 10
ansys.Format('', 'E', nsigfig + 9, nsigfig)
ansys.Page(1E9, '', -1, 240)
except: # for travis and appveyor
pass
def test_prnsol_u(self):
# verify cyclic displacements
table = self.ansys.Prnsol('u').splitlines()
if self.ansys.using_corba:
array = np.genfromtxt(table[7:])
else:
array = np.genfromtxt(table[9:])
ansys_nnum = array[:, 0].astype(np.int)
ansys_disp = array[:, 1:-1]
nnum, disp = self.result.nodal_solution(0)
# cyclic model will only output the master sector
ansys_nnum = ansys_nnum[:nnum.size]
ansys_disp = ansys_disp[:nnum.size]
assert np.allclose(ansys_nnum, nnum)
assert np.allclose(ansys_disp, disp)
def test_presol_s(self):
# verify element stress
element_stress, elemnum, enode = self.result.element_stress(0)
element_stress = np.vstack(element_stress)
enode = np.hstack(enode)
# parse ansys result
table = self.ansys.Presol('S').splitlines()
ansys_element_stress = []
line_length = len(table[15])
for line in table:
if len(line) == line_length:
ansys_element_stress.append(line)
ansys_element_stress = np.genfromtxt(ansys_element_stress)
ansys_enode = ansys_element_stress[:, 0].astype(np.int)
ansys_element_stress = ansys_element_stress[:, 1:]
assert np.allclose(element_stress, ansys_element_stress)
assert np.allclose(enode, ansys_enode)
def test_prnsol_s(self):
# verify cyclic displacements
table = self.ansys.Prnsol('s').splitlines()
if self.ansys.using_corba:
array = np.genfromtxt(table[7:])
else:
array = np.genfromtxt(table[10:])
ansys_nnum = array[:, 0].astype(np.int)
ansys_stress = array[:, 1:]
nnum, stress = self.result.nodal_stress(0)
# v150 includes nodes in the geometry that aren't in the result
mask = np.in1d(nnum, ansys_nnum)
nnum = nnum[mask]
stress = stress[mask]
assert np.allclose(ansys_nnum, nnum)
assert np.allclose(ansys_stress, stress)
def test_prnsol_prin(self):
# verify principal stress
table = self.ansys.Prnsol('prin').splitlines()
if self.ansys.using_corba:
array = np.genfromtxt(table[7:])
else:
array = np.genfromtxt(table[10:])
ansys_nnum = array[:, 0].astype(np.int)
ansys_stress = array[:, 1:]
nnum, stress = self.result.principal_nodal_stress(0)
# v150 includes nodes in the geometry that aren't in the result
mask = np.in1d(nnum, ansys_nnum)
nnum = nnum[mask]
stress = stress[mask]
assert np.allclose(ansys_nnum, nnum)
assert np.allclose(ansys_stress, stress, atol=1E-2)
@pytest.mark.skipif(not running_xserver(), reason="Requires active X Server")
def test_plot(self):
filename = '/tmp/temp.png'
self.result.plot_nodal_solution(0, screenshot=filename,
interactive=False)
# self.result.plot_nodal_stress(0, 'Sx', screenshot=filename,
# interactive=False)
self.result.plot_principal_nodal_stress(0, 'SEQV', screenshot=filename,
interactive=False)
def test_exit(self):
self.ansys.Exit()
