initial upload

PyLaGriT/tests/.directory

@@ -0,0 +1,3 @@
+[Dolphin]
+Timestamp=2014,7,24,15,5,28
+ViewMode=1

PyLaGriT/tests/test_pylagrit.py

@@ -0,0 +1,204 @@
+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)
+    
+    
+    

PyLaGriT/tests/test_scripts/test_script1.lg

@@ -0,0 +1 @@
+cmo / create / cmotri / / / tri

PyLaGriT/tests/test_scripts/test_script2.lg

@@ -0,0 +1,6 @@
+*Example Comment
+cmo / create / cmotri / / / tri
+read / avs / input.inp / cmotri
+*
+
+finish

PyLaGriT/tests/test_scripts/test_script3.lg

@@ -0,0 +1,8 @@
+* Example of Some Comments
+* More Comments
+* 
+define/nx/20
+define/ny/20
+define/nz/20
+   
+finish