wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-06-28 14:12:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

minor updates in interpolation in TensorMesh

Browse Source
This commit is contained in:
rowanc1
2014-04-15 15:12:19 -07:00
parent c39163bd6e
commit dd579ddbd4
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Mesh/TensorMesh.py
+5 -1
View File
@@ -220,8 +220,12 @@ class BaseTensorMesh(BaseRectangularMesh):
if type(loc) == list:
loc = np.array(loc)
assert type(loc) == np.ndarray, "must be a numpy array"
if loc.size == self.dim:
loc = np.atleast_2d(loc)
if self.dim > 1:
assert loc.shape[1] == self.dim, "must be a column vector of shape (nPts, mesh.dim)"
assert loc.shape[1] == self.dim, "must be a column vector of shape (nPts, mesh.dim) not (%d,%d)" % loc.shape
elif len(loc.shape) == 1:
loc = loc[:,np.newaxis]
else: