reworked class definition.
now follows workflow of 1. clean 2. prepare 3. run 4. postprocess 5. compare/update added file-structural helper functions.
This commit is contained in:
Philip Eisenlohr
parent
aba4703463
commit
0b4b154abb
|
|
@ -6,155 +6,115 @@ import subprocess,shutil
|
|||
import damask
|
||||
|
||||
class Test():
|
||||
'''
|
||||
General class for testing.
|
||||
Is sub-classed by the individual tests.
|
||||
'''
|
||||
|
||||
variants = []
|
||||
|
||||
def __init__(self):
|
||||
self.dirBase = os.path.dirname(os.path.realpath(sys.argv[0]))
|
||||
|
||||
def execute(self,variants = [],update = []):
|
||||
'''
|
||||
General class for testing.
|
||||
Is sub-classed by the individual tests.
|
||||
Run all variants and report first failure.
|
||||
'''
|
||||
#define those according to your test
|
||||
modelname=None
|
||||
jobname=None
|
||||
test_dir=None
|
||||
spectral_options=None
|
||||
compile=False = None
|
||||
post_txt = None
|
||||
tol = 0.0
|
||||
|
||||
has_variants=False # False ==> A single test case is run
|
||||
#has_variants=True # True ==> Need to define the test_variants generator
|
||||
if len(variants) == 0: variants = xrange(len(self.variants)) # iterate over all variants
|
||||
|
||||
def test_variants(self):
|
||||
'''
|
||||
If has_subtests == True this method defines the generator for test variants
|
||||
This generator must be defined in each test,
|
||||
depending on what to change: orientations, parameters,....
|
||||
Below is an EXAMPLE.
|
||||
'''
|
||||
#maybe get rid of has_variants through testing for None return value...
|
||||
yield(None)
|
||||
self.clean()
|
||||
for variant in variants:
|
||||
self.prepare(variant)
|
||||
self.run(variant)
|
||||
self.postprocess(variant)
|
||||
if variant in update:
|
||||
self.update(variant)
|
||||
elif not self.compare(variant):
|
||||
return variant
|
||||
return -1
|
||||
|
||||
def test_variants_example(self):
|
||||
subtest_orientations=[[0.0,90.0,0.0],[0.0,0.0,90.0]]
|
||||
for i,o in enumerate(subtest_orientations):
|
||||
from damask_tools import MATERIAL_CONFIG
|
||||
mat=MATERIAL_CONFIG()
|
||||
mat.read('material.config_base')
|
||||
mat.add_texture(label='Grain001',
|
||||
type ='(gauss)',
|
||||
eulers = o)
|
||||
mat.write(overwrite=True)
|
||||
print(mat.data['texture']['Grain001'])
|
||||
testlabel='orientation_%03i'%i
|
||||
yield(testlabel)
|
||||
|
||||
def run_test(self):
|
||||
res=[]
|
||||
if self.has_variants:
|
||||
for t in self.test_variants():
|
||||
print '###############################################'
|
||||
print '###############################################'
|
||||
print(t)
|
||||
print '###############################################'
|
||||
val=self.run_single_test(t)
|
||||
res.append(val==True)
|
||||
else:
|
||||
val=self.run_single_test()
|
||||
res.append(val==True)
|
||||
if all(res) is True:
|
||||
return True
|
||||
print(res)
|
||||
return False
|
||||
|
||||
def run_single_test(self,variant):
|
||||
self.clean_current_results()
|
||||
if self.calc_current_results(variant) is False:
|
||||
return False
|
||||
print('simulation finished')
|
||||
self.postprocess()
|
||||
if self.compare_to_reference() is False:
|
||||
print '++++++++ Test not OK +++++++++'
|
||||
return False
|
||||
print 'Test OK'
|
||||
return True
|
||||
def clean(self):
|
||||
'''
|
||||
Delete directory tree containing current results.
|
||||
'''
|
||||
status = True
|
||||
|
||||
def clean_current_results(self):
|
||||
os.chdir(self.test_dir)
|
||||
try:
|
||||
shutil.rmtree('current_results')
|
||||
except:
|
||||
print('Could not delete current_results')
|
||||
os.mkdir('current_results')
|
||||
try:
|
||||
shutil.rmtree(self.dirCurrent())
|
||||
except:
|
||||
print('removal of directory "%s" failed...'%(self.dirCurrent()))
|
||||
status = status and False
|
||||
|
||||
def calc_current_results(self):
|
||||
'''
|
||||
Should be defined in the individual tests
|
||||
'''
|
||||
pass
|
||||
try:
|
||||
os.mkdir(self.dirCurrent())
|
||||
except:
|
||||
print('creation of directory "%s" failed...'%(self.dirCurrent()))
|
||||
status = status and False
|
||||
|
||||
def calc_marc(self,compile=None):
|
||||
'''
|
||||
Calculates current results for MSC.Marc
|
||||
'''
|
||||
if compile is None: compile=self.compile
|
||||
self.copy_from_ref_list()
|
||||
self.copy_files_from_reference_results()
|
||||
os.chdir('./current_results')
|
||||
m = damask.solver.Marc()
|
||||
m.submit_job(compile=compile, compiled_dir='../../../code/')
|
||||
print('simulation submitted')
|
||||
self.exit_number = m.exit_number_from_outFile(outFile=self.modelname+'_'+self.jobname+'.out')
|
||||
|
||||
if not self.exit_number==3004:
|
||||
print('Job did not exit with No. 3004')
|
||||
return False
|
||||
return True
|
||||
|
||||
def calc_spectral(self, compile=None):
|
||||
pass
|
||||
return status
|
||||
|
||||
def copy_from_ref_list(self):
|
||||
self.copy_from_ref=[self.modelname+'_'+self.jobname+'.dat',
|
||||
self.modelname+'.mfd', # for dev
|
||||
'material.config',
|
||||
]
|
||||
|
||||
|
||||
def copy_files_from_reference_results(self):
|
||||
for file in self.copy_from_ref:
|
||||
shutil.copy2('./reference_results/%s'%file,'./current_results/%s'%file)
|
||||
# Note: possibly symlinking? No, because copy is OS independent.
|
||||
|
||||
def read_val_from_file(self,fname=None):
|
||||
file = open(fname,'r')
|
||||
headerlength = int(file.readline().split()[0]) + 1
|
||||
file.close
|
||||
import numpy as N
|
||||
val = N.loadtxt(fname,skiprows=headerlength)
|
||||
return val
|
||||
def prepare(self,variant):
|
||||
'''
|
||||
Do all necessary preparations for the run of each test variant
|
||||
'''
|
||||
return True
|
||||
|
||||
|
||||
def compare_to_reference(self):
|
||||
import string, numpy as N
|
||||
print 'comparing results against reference_results...'
|
||||
os.chdir(os.path.join(self.test_dir,'current_results'))
|
||||
cur=self.read_val_from_file(fname='postProc/'+self.post_txt)
|
||||
ref=self.read_val_from_file(fname='../reference_results/postProc/'+self.post_txt)
|
||||
|
||||
err=abs((ref/cur)-1.) # relative tolerance
|
||||
#err=abs(ref-cur) # absolute tolerance
|
||||
print 'tol', self.tol
|
||||
print 'max error', N.max(err)
|
||||
if N.max(err)>self.tol:
|
||||
return False
|
||||
return True
|
||||
|
||||
def postprocess(self):
|
||||
print 'postprocessing results ...'
|
||||
os.chdir(self.test_dir)
|
||||
file=open('./postprocessing.cmd','r')
|
||||
postproc=file.readlines()
|
||||
file.close()
|
||||
os.chdir(os.path.join(self.test_dir,'current_results'))
|
||||
for cmd in postproc: # PHILIP: suggestion to just execute the script "postprocessing" directly within a shell, i.e. os.system('../postprocessing')
|
||||
print(cmd)
|
||||
os.system(cmd) # PHILIP: reason is that for loops and the like get broken with line by line execution from here...
|
||||
def run(self,variant):
|
||||
'''
|
||||
Execute the requested test variant.
|
||||
'''
|
||||
return True
|
||||
|
||||
|
||||
def postprocess(self,variant):
|
||||
'''
|
||||
Perform post-processing of generated results for this test variant.
|
||||
'''
|
||||
return True
|
||||
|
||||
|
||||
def compare(self,variant):
|
||||
'''
|
||||
Compare reference to current results.
|
||||
'''
|
||||
return True
|
||||
|
||||
|
||||
def update(self,variant):
|
||||
'''
|
||||
Update reference with current results.
|
||||
'''
|
||||
return True
|
||||
|
||||
|
||||
def dirReference(self):
|
||||
'''
|
||||
'''
|
||||
return os.path.normpath(os.path.join(self.dirBase,'reference/'))
|
||||
|
||||
def dirCurrent(self):
|
||||
'''
|
||||
'''
|
||||
return os.path.normpath(os.path.join(self.dirBase,'current/'))
|
||||
|
||||
def fileInReference(self,file):
|
||||
'''
|
||||
'''
|
||||
return os.path.join(self.dirReference(),file)
|
||||
|
||||
def fileInCurrent(self,file):
|
||||
'''
|
||||
'''
|
||||
return os.path.join(self.dirCurrent(),file)
|
||||
|
||||
def copy_Reference2Current(self,files=[]):
|
||||
for file in files:
|
||||
shutil.copy2(self.fileInReference(file),self.fileInCurrent(file))
|
||||
|
||||
def copy_Current2Current(self,files=[]):
|
||||
for file in files:
|
||||
shutil.copy2(self.fileInCurrent(file[0]),self.fileInCurrent(file[1]))
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue