Skip to content
Snippets Groups Projects
Commit 7079da98 authored by Rahil Doshi's avatar Rahil Doshi
Browse files

Update TestArrayContainer.py to generate class SS316L for test_interpolation.cpp

parent a80aa59e
Branches
Tags
No related merge requests found
import numpy as np import numpy as np
from pystencils import fields from pystencils import fields
from importlib.resources import files
from pystencilssfg import SourceFileGenerator from pystencilssfg import SourceFileGenerator
from pymatlib.data.alloys.SS316L import SS316L from pymatlib.core.yaml_parser import create_alloy_from_yaml
from pymatlib.core.interpolators import InterpolationArrayContainer from pymatlib.core.interpolators import InterpolationArrayContainer
with SourceFileGenerator() as sfg: with SourceFileGenerator() as sfg:
'''u = fields(f"u: double[2D]", layout='fzyx') u = fields(f"u: double[2D]", layout='fzyx')
mat = SS316L.create_SS316L(u.center())
arr_container = DoubleLookupArrayContainer.from_material("SS316L", mat)
sfg.generate(arr_container)'''
T = np.array([3243.15, 3253.15, 3263.15, 3273.15], dtype=np.float64) T_bs = np.array([3243.15, 3248.15, 3258.15, 3278.15], dtype=np.float64)
E = np.array([1.68e10, 1.69e10, 1.70e10, 1.71e10], dtype=np.float64) E_bs = np.array([1.68e10, 1.69e10, 1.70e10, 1.71e10], dtype=np.float64)
custom_container = InterpolationArrayContainer("DoubleLookupTests", np.flip(T), np.flip(E)) custom_container = InterpolationArrayContainer("BinarySearchTests", T_bs, E_bs)
sfg.generate(custom_container) sfg.generate(custom_container)
custom_container = InterpolationArrayContainer("BinarySearchTests", T, E) T_eq = np.array([3243.15, 3253.15, 3263.15, 3273.15], dtype=np.float64)
sfg.generate(custom_container) E_neq = np.array([1.68e10, 1.69e10, 1.70e10, 1.71e10], dtype=np.float64)
\ No newline at end of file
custom_container = InterpolationArrayContainer("DoubleLookupTests", np.flip(T_eq), np.flip(E_neq))
sfg.generate(custom_container)
yaml_path = files('pymatlib.data.alloys.SS316L').joinpath('SS304L.yaml')
mat = create_alloy_from_yaml(yaml_path, u.center())
arr_container = InterpolationArrayContainer.from_material("SS316L", mat)
sfg.generate(arr_container)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment