def setUp(self):
url = "https://storage.googleapis.com/simpeg/tests/dc_utils/"
cloudfiles = [
"mesh3d.msh",
"2spheres_conmodel.npy",
"rhoA_GIF_dd.txt",
"rhoA_GIF_dp.txt",
"rhoA_GIF_pd.txt",
"rhoA_GIF_pp.txt",
]
self.basePath = os.path.expanduser("~/Downloads/TestDCUtilsHalfSpace")
io_utils.download(
[url + f for f in cloudfiles], folder=self.basePath, overwrite=True
)
# Load Mesh
mesh_file = os.path.sep.join([self.basePath, "mesh3d.msh"])
mesh = discretize.load_mesh(mesh_file)
self.mesh = mesh
# Load Model
model_file = os.path.sep.join([self.basePath, "2spheres_conmodel.npy"])
model = np.load(model_file)
self.model = model
xmin, xmax = -15.0, 15.0
ymin, ymax = 0.0, 0.0
zmin, zmax = -0.25, -0.25
xyz = np.array([[xmin, ymin, zmin], [xmax, ymax, zmax]])
self.xyz = xyz
self.survey_a = 1.0
self.survey_b = 1.0
self.survey_n = 10
self.plotIt = False
def test_save_load(self):
print('\nTesting save / load of Cyl Mesh ...')
mesh0 = self.mesh
f = mesh0.save()
mesh1 = discretize.load_mesh(f)
compare_meshes(self, mesh0, mesh1)
os.remove(f)
# unzip the tarfile
tar = tarfile.open(downloads, "r")
tar.extractall()
tar.close()
# Download the inverted model
f = discretize.utils.download(
"https://storage.googleapis.com/simpeg/laguna_del_maule_slicer.tar.gz",
overwrite=True,
)
tar = tarfile.open(f, "r")
tar.extractall()
tar.close()
# Load the mesh/data
mesh = discretize.load_mesh(
os.path.join("laguna_del_maule_slicer", "mesh.json"))
models = {
"Lpout": np.load(os.path.join("laguna_del_maule_slicer", "Lpout.npy"))
}
###############################################################################
# Create PyVista data objects
# ---------------------------
#
# Here we start making PyVista data objects of all the spatially referenced
# data.
# Get the PyVista dataset of the inverted model
dataset = mesh.to_vtk(models)
###############################################################################
