205 lines
6.2 KiB
Python
205 lines
6.2 KiB
Python
|
import unittest
|
||
|
|
import pylagrit
|
||
|
|
import glob
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
from contextlib import contextmanager
|
||
|
|
import itertools
|
||
|
|
|
||
|
|
class TestPyLaGriT(unittest.TestCase):
|
||
|
|
'''
|
||
|
|
A PyLagriT Test
|
||
|
|
|
||
|
|
Represents a test of PyLaGriT functionality.
|
||
|
|
'''
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
#Sets upt a lagrit object to be used during tests.
|
||
|
|
with suppress_stdout():
|
||
|
|
self.lg = pylagrit.PyLaGriT('../../build/lagrit')
|
||
|
|
|
||
|
|
def test_read_script(self):
|
||
|
|
'''
|
||
|
|
Test the Read Script Function
|
||
|
|
|
||
|
|
Tests that pylagrit can read a lagrit script.
|
||
|
|
'''
|
||
|
|
|
||
|
|
lg = self.lg
|
||
|
|
for f in glob.glob('/test_scripts/*.lg'):
|
||
|
|
with suppress_stdout():
|
||
|
|
lg.read_script(f)
|
||
|
|
|
||
|
|
def test_convert(self):
|
||
|
|
'''
|
||
|
|
Test the Conversion Function
|
||
|
|
|
||
|
|
Tests that pylagrit can convert files from one format to a new one.
|
||
|
|
'''
|
||
|
|
|
||
|
|
lg = self.lg
|
||
|
|
|
||
|
|
old_formats = ['avs', 'gmv']
|
||
|
|
new_formats = ['avs', 'gmv', 'exo']
|
||
|
|
|
||
|
|
for old_ft, new_ft in itertools.product(old_formats, new_formats):
|
||
|
|
#Convert all avs files to gmv with pylagrit
|
||
|
|
with suppress_stdout():
|
||
|
|
lg.convert('test_convert/*.%s'%old_ft, new_ft)
|
||
|
|
|
||
|
|
#If the file did not get generated, assume fail.
|
||
|
|
if not os.path.isfile('test.%s'%new_ft):
|
||
|
|
raise OSError('Failed Conversion.')
|
||
|
|
os.remove('test.%s'%new_ft)
|
||
|
|
|
||
|
|
def test_merge(self):
|
||
|
|
'''
|
||
|
|
Tests the Merge Function
|
||
|
|
|
||
|
|
Tests that pylagrit can merge 3 mesh objects and return a single mesh
|
||
|
|
object.
|
||
|
|
'''
|
||
|
|
|
||
|
|
lg = self.lg
|
||
|
|
with suppress_stdout():
|
||
|
|
mo1 = lg.read('avs', 'contour_file.avs')
|
||
|
|
mo2 = lg.read('avs', 'contour_file.avs')
|
||
|
|
mo3 = lg.read('avs', 'contour_file.avs')
|
||
|
|
new_mo = lg.merge([mo1, mo2, mo3])
|
||
|
|
#Test that the merge created a new mesh object.
|
||
|
|
if type(new_mo) is type(None):
|
||
|
|
raise ValueError('An expected mesh object was not created.')
|
||
|
|
|
||
|
|
def test_create(self):
|
||
|
|
'''
|
||
|
|
Tests the Create Function
|
||
|
|
|
||
|
|
Tests that the create function returns a mesh object.
|
||
|
|
'''
|
||
|
|
|
||
|
|
lg = self.lg
|
||
|
|
with suppress_stdout():
|
||
|
|
mo1 = lg.create()
|
||
|
|
mo2 = lg.create_hex()
|
||
|
|
mo3 = lg.create(elem_type='pri')
|
||
|
|
if any([type(x) is type(None) for x in [mo1, mo2, mo3]]):
|
||
|
|
raise ValueError('An expected mesh object was not created.')
|
||
|
|
|
||
|
|
def test_createpts_brick(self):
|
||
|
|
'''
|
||
|
|
Test the Create Points Function
|
||
|
|
|
||
|
|
Tests that a mesh object can run the createpts method with fail.
|
||
|
|
'''
|
||
|
|
|
||
|
|
lg = self.lg
|
||
|
|
with suppress_stdout():
|
||
|
|
mo = lg.create()
|
||
|
|
npts = (4, 4, 4)
|
||
|
|
mins = (0, 0, 0)
|
||
|
|
maxs = (4, 4, 4)
|
||
|
|
mo.createpts_brick('xyz', npts, mins, maxs)
|
||
|
|
|
||
|
|
def test_pset_geom(self):
|
||
|
|
'''
|
||
|
|
Test the MO Geometry Function
|
||
|
|
|
||
|
|
Tests that each geom method returns a PSet.
|
||
|
|
'''
|
||
|
|
|
||
|
|
lg = self.lg
|
||
|
|
mins = (1, 1, 1)
|
||
|
|
maxs = (2, 2, 2)
|
||
|
|
with suppress_stdout():
|
||
|
|
mo = lg.create()
|
||
|
|
ps = []
|
||
|
|
ps.append(mo.pset_geom(mins, maxs))
|
||
|
|
ps.append(mo.pset_geom_xyz(mins, maxs))
|
||
|
|
ps.append(mo.pset_geom_rtz(mins, maxs))
|
||
|
|
ps.append(mo.pset_geom_rtp(mins, maxs))
|
||
|
|
|
||
|
|
if any([not isinstance(x, pylagrit.PSet) for x in ps]):
|
||
|
|
raise ValueError('PSet not returned.')
|
||
|
|
|
||
|
|
def test_copy(self):
|
||
|
|
'''
|
||
|
|
Test the Copy Function
|
||
|
|
|
||
|
|
Tests that the copy function returns a mesh object.
|
||
|
|
'''
|
||
|
|
|
||
|
|
lg = self.lg
|
||
|
|
with suppress_stdout():
|
||
|
|
mo = lg.create()
|
||
|
|
mo_clone = lg.copy(mo)
|
||
|
|
if not isinstance(mo_clone, pylagrit.MO):
|
||
|
|
raise ValueError('MO not returned.')
|
||
|
|
|
||
|
|
def test_pset_not(self):
|
||
|
|
'''
|
||
|
|
Test the Mesh Object Not Function
|
||
|
|
|
||
|
|
Tests that a mesh object can return a PSet using the pset_not function.
|
||
|
|
'''
|
||
|
|
|
||
|
|
lg = self.lg
|
||
|
|
mins = (1, 1, 1)
|
||
|
|
maxs = (2, 2, 2)
|
||
|
|
with suppress_stdout():
|
||
|
|
mo = lg.create()
|
||
|
|
ps1 = mo.pset_geom(mins, maxs)
|
||
|
|
ps2 = mo.pset_not(ps1)
|
||
|
|
if not isinstance(ps2, pylagrit.PSet):
|
||
|
|
raise ValueError('PSet not returned.')
|
||
|
|
|
||
|
|
def test_subset(self):
|
||
|
|
'''
|
||
|
|
Test the MO Subset Function
|
||
|
|
|
||
|
|
Tests that a mesh object can return another mesh object with the subset
|
||
|
|
methods.
|
||
|
|
'''
|
||
|
|
|
||
|
|
lg = self.lg
|
||
|
|
mins = (1,1,1)
|
||
|
|
maxs = (2,2,2)
|
||
|
|
with suppress_stdout():
|
||
|
|
mo = lg.create()
|
||
|
|
mo_subs = []
|
||
|
|
mo_subs = mo_subs + [mo.subset(mins, maxs)]
|
||
|
|
mo_subs = mo_subs + [mo.subset(mins, maxs, geom='rtz')]
|
||
|
|
mo_subs = mo_subs + [mo.subset(mins, maxs, geom='rtp')]
|
||
|
|
mo_subs = mo_subs + [mo.subset_rtp(mins, maxs)]
|
||
|
|
mo_subs = mo_subs + [mo.subset_rtz(mins, maxs)]
|
||
|
|
|
||
|
|
if any([not isinstance(x, pylagrit.MO) for x in mo_subs]):
|
||
|
|
raise ValueError('MO not returned.')
|
||
|
|
|
||
|
|
@contextmanager
|
||
|
|
def suppress_stdout():
|
||
|
|
#Utility to supress standard output.
|
||
|
|
with open(os.devnull, 'w') as devnull:
|
||
|
|
old_stdout = sys.stdout
|
||
|
|
sys.stdout = devnull
|
||
|
|
try:
|
||
|
|
yield
|
||
|
|
finally:
|
||
|
|
sys.stdout = old_stdout
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
runner = unittest.TextTestRunner(verbosity=2)
|
||
|
|
suite = unittest.TestSuite()
|
||
|
|
suite.addTest(TestPyLaGriT('test_read_script'))
|
||
|
|
suite.addTest(TestPyLaGriT('test_convert'))
|
||
|
|
suite.addTest(TestPyLaGriT('test_merge'))
|
||
|
|
suite.addTest(TestPyLaGriT('test_create'))
|
||
|
|
suite.addTest(TestPyLaGriT('test_createpts_brick'))
|
||
|
|
suite.addTest(TestPyLaGriT('test_pset_geom'))
|
||
|
|
suite.addTest(TestPyLaGriT('test_copy'))
|
||
|
|
suite.addTest(TestPyLaGriT('test_pset_not'))
|
||
|
|
suite.addTest(TestPyLaGriT('test_subset'))
|
||
|
|
runner.run(suite)
|
||
|
|
|
||
|
|
|
||
|
|
|