Homec4science
Diffusion LibMultiScale (features/coupler_component) Browse

libmultiscale/test/single_scale/AKANTU/test_static/check_strain.py
0a2b48144aacfeatures/coupler_compon…

Last Change
Raw File

check_strain.py
View Options

#!/usr/bin/env python
import numpy as np
import sys
from LMUnitsModule import *
from apply_strain import shift, length
disp_name = "disp_extr"
name = "all_pos"
tol_mean = 1e-8
tol_max = 1e-7
def compute( **kwargs):
keys = kwargs.keys()
if name in keys:
key = name
try:
pos0 = kwargs[key]
except UnboundLocalError as err:
raise UnboundLocalError(
"{0}\nThe only available keys are {1}. The key '{2}' wasn't found!".format(err, keys, name))
if pos0.shape[0] == 0:
return pos0
print (pos0.shape)
if disp_name in keys:
key = disp_name
try:
real_disp = kwargs[key]
except UnboundLocalError as err:
raise UnboundLocalError(
"{0}\nThe only available keys are {1}. The key '{2}' wasn't found!".format(err, keys, disp_name))
grad_u = np.matrix([[0, shift[0]/length, 0],
[0, shift[1]/length, 0],
[0, shift[2]/length, 0]])
theoretical_disp = np.array(np.matrix(pos0)*grad_u)
try:
real_diff = theoretical_disp - real_disp
except ValueError as err:
raise ValueError(
"{0}\n theoretical_disp.shape = {1}\n real_disp.shape = {2}".format(
err, theoretical_disp.shape, real_disp.shape));
print("Real {0}, Th {1}".format(real_disp.shape, theoretical_disp.shape))
dists = (real_diff**2).sum(1)**.5
max_err = dists.max()
mean_err = dists.mean()
print ("\n")
print ("the tolerances on mean and max displacement "
"are {0} and {1}, respectively".format(tol_mean, tol_max))
def symbol(val, tol):
if val < tol:
return "<"
elif val == tol:
return "="
else:
return ">"
mean_ok = mean_err < tol_mean
max_ok = max_err < tol_max
print ("the maximum error on displacement = {0} {1} tol: {2}".format(
max_err, symbol(max_err, tol_max),
"passed" if max_ok else "FAILED"))
print ("the mean error on displacement = {0} {1} tol: {2}".format(
mean_err, symbol(mean_err, tol_mean),
"passed" if mean_ok else "FAILED"))
if max_ok and mean_ok:
print("Test succesful")
lm_exit(0)
return pos0
print("the theoretical displacements are:\n{0}".format(theoretical_disp))
print("the measured displacements are:\n{0}".format(real_disp))
print("Test failed")
lm_exit(2)
return pos0
c4science · Help