diff --git a/src/pymatlib/core/cpp/TestArrayContainer.py b/src/pymatlib/core/cpp/TestArrayContainer.py
index af9fe9edd461ced6da472345e30bbf1af7390b76..a20d260637c211345334af2fed0b3be8d708f24d 100644
--- a/src/pymatlib/core/cpp/TestArrayContainer.py
+++ b/src/pymatlib/core/cpp/TestArrayContainer.py
@@ -1,21 +1,27 @@
 import numpy as np
 from pystencils import fields
+from importlib.resources import files
 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
 
 
 with SourceFileGenerator() as sfg:
-    '''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)'''
+    u = fields(f"u: double[2D]", layout='fzyx')
 
-    T = np.array([3243.15, 3253.15, 3263.15, 3273.15], dtype=np.float64)
-    E = np.array([1.68e10, 1.69e10, 1.70e10, 1.71e10], dtype=np.float64)
+    T_bs = np.array([3243.15, 3248.15, 3258.15, 3278.15], 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)
 
-    custom_container = InterpolationArrayContainer("BinarySearchTests", T, E)
-    sfg.generate(custom_container)
\ No newline at end of file
+    T_eq = np.array([3243.15, 3253.15, 3263.15, 3273.15], dtype=np.float64)
+    E_neq = np.array([1.68e10, 1.69e10, 1.70e10, 1.71e10], dtype=np.float64)
+
+    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)