Update structs_10 to nested structs containing array members

Author: czgdp1807

integration_tests/structs_10.py

@@ -2,17 +2,25 @@
 from numpy import empty, float64
 
 @dataclass
-class MatVec:
+class Mat:
     mat: f64[2, 2]
+
+@dataclass
+class Vec:
     vec: f64[2]
 
+@dataclass
+class MatVec:
+    mat: Mat
+    vec: Vec
+
 def rotate(mat_vec: MatVec) -> f64[2]:
     rotated_vec: f64[2] = empty(2, dtype=float64)
-    rotated_vec[0] = mat_vec.mat[0, 0] * mat_vec.vec[0] + mat_vec.mat[0, 1] * mat_vec.vec[1]
-    rotated_vec[1] = mat_vec.mat[1, 0] * mat_vec.vec[0] + mat_vec.mat[1, 1] * mat_vec.vec[1]
+    rotated_vec[0] = mat_vec.mat.mat[0, 0] * mat_vec.vec.vec[0] + mat_vec.mat.mat[0, 1] * mat_vec.vec.vec[1]
+    rotated_vec[1] = mat_vec.mat.mat[1, 0] * mat_vec.vec.vec[0] + mat_vec.mat.mat[1, 1] * mat_vec.vec.vec[1]
     return rotated_vec
 
-def test_rotate_by_90():
+def create_MatVec_obj() -> MatVec:
     mat: f64[2, 2] = empty((2, 2), dtype=float64)
     vec: f64[2] = empty(2, dtype=float64)
     mat[0, 0] = 0.0
@@ -21,9 +29,15 @@ def test_rotate_by_90():
     mat[1, 1] = 0.0
     vec[0] = 1.0
     vec[1] = 0.0
-    mat_vec: MatVec = MatVec(mat, vec)
-    print(mat_vec.mat[0, 0], mat_vec.mat[0, 1], mat_vec.mat[1, 0], mat_vec.mat[1, 1])
-    print(mat_vec.vec[0], mat_vec.vec[1])
+    mat_s: Mat = Mat(mat)
+    vec_s: Vec = Vec(vec)
+    mat_vec: MatVec = MatVec(mat_s, vec_s)
+    return mat_vec
+
+def test_rotate_by_90():
+    mat_vec: MatVec = create_MatVec_obj()
+    print(mat_vec.mat.mat[0, 0], mat_vec.mat.mat[0, 1], mat_vec.mat.mat[1, 0], mat_vec.mat.mat[1, 1])
+    print(mat_vec.vec.vec[0], mat_vec.vec.vec[1])
     rotated_vec: f64[2] = rotate(mat_vec)
     print(rotated_vec[0], rotated_vec[1])
     assert abs(rotated_vec[0] - 0.0) <= 1e-12