fix: all tests passing

Author: lachlangrose

LoopStructural/interpolators/supports/_3d_structured_grid.py

@@ -141,8 +141,6 @@ def position_to_dof_coefs(self, pos):
         weights = self.trilinear(x_local, y_local, local_z)
         return weights
 
-   
-
     def neighbour_global_indexes(self, mask=None, **kwargs):
         """
         Get neighbour indexes
@@ -458,8 +456,11 @@ def get_element_for_location(self, pos):
             [description]
         """
         vertices, inside = self.position_to_cell_vertices(pos)
+        vertices = np.array(vertices)
+        vertices = vertices.reshape((vertices.shape[1], 8, 3))
         elements, inside = self.position_to_cell_corners(pos)
         a = self.position_to_dof_coefs(pos)
         return vertices, a.T, elements, inside
+
     def get_elements(self):
-        return 
+        return

tests/unit/interpolator/test_discrete_supports.py

@@ -96,9 +96,9 @@ def test_evaluate_gradient2(support_class):
 def test_get_element(support):
     point = support.barycentre[[0], :]
     vertices, dof, idc, inside = support.get_element_for_location(point)
-    print(vertices)
+    # vertices = vertices.reshape(-1, 3)
     bary = np.mean(vertices, axis=1)
-    assert np.sum(point - bary) == 0
+    assert np.isclose(np.sum(point - bary), 0)
 
 
 def test_global_to_local_coordinates():