py - refactor tests needing output

python/build_ceed_cffi.py

@@ -36,6 +36,7 @@
     header = '\n'.join(lines)
     header = header.split("static inline CeedInt CeedIntPow", 1)[0]
     header += '\nextern int CeedVectorGetState(CeedVector, uint64_t*);'
+    header += '\nextern int CeedElemRestrictionGetELayout(CeedElemRestriction, CeedInt *layout);'
     # Note: cffi cannot handle vargs
     header = re.sub("va_list", "const char *", header)
 ffibuilder.cdef(header)

python/ceed_elemrestriction.py

@@ -131,6 +131,29 @@ def get_multiplicity(self):
         # Return
         return mult
 
+    # Get ElemRestrition Layout
+    def get_layout(self):
+        """Get the element vector layout of an ElemRestriction.
+
+           Returns:
+             layout: Vector containing layout array, stored as [nodes, components, elements].
+                     The data for node i, component j, element k in the element
+                     vector is given by i*layout[0] + j*layout[1] + k*layout[2]."""
+
+        # Create output array
+        layout = np.zeros(3, dtype="int32")
+        array_pointer = ffi.cast(
+            "CeedInt *",
+            layout.__array_interface__['data'][0])
+
+        # libCEED call
+        err_code = lib.CeedElemRestrictionGetELayout(
+            self._pointer[0], array_pointer)
+        self._ceed._check_error(err_code)
+
+        # Return
+        return layout
+
 # ------------------------------------------------------------------------------