[ENH] UFEM: Add 3D assembly ptest