def test_plot_categorical(self):
'''
Test the capability of plotting along xz/yz orientation and use line plots based on drill hole id for a categorical variable
'''
from matplotlib import pyplot as plt
output_file = os.path.join(self.out_dir, 'location_plot3.png')
_, ax = plt.subplots(1, 1)
try:
_ = gs.location_plot(self.data,
var='Lithofacies',
orient='yz',
aspect=5,
plot_collar=True,
output_file=output_file)
_ = gs.location_plot(self.data,
var='Lithofacies',
orient='yz',
aspect=5,
plot_collar=True,
output_file=output_file,
ax=ax)
except Exception as ex:
self.fail(
'Unable to test the location plot along yz/xz orientation for a categorical variable \n{}'
.format(str(ex)))
self.assertEqual(os.path.isfile(output_file), True)
def test_plot_cmap(self):
'''
Test the capability of plotting by passing variable name to get the color map
'''
from matplotlib import pyplot as plt
output_file = os.path.join(self.out_dir, 'location_plot2.png')
_, ax = plt.subplots(1, 1)
try:
_ = gs.location_plot(self.data, var='Phi', output_file=output_file)
_ = gs.location_plot(self.data,
var='Phi',
output_file=output_file,
ax=ax)
except Exception as ex:
self.fail(
'Unable to test the location plot with passing the variable name \n{}'
.format(str(ex)))
self.assertEqual(os.path.isfile(output_file), True)
def test_simple_plot(self):
'''
Test the capability of plotting by passing only the data file
'''
output_file = os.path.join(self.out_dir, 'location_plot1.png')
try:
_ = gs.location_plot(self.data, output_file=output_file)
except Exception as ex:
self.fail(
'Unable to test the location plot with simple settings \n{}'.
format(str(ex)))
self.assertEqual(os.path.isfile(output_file), True)
def test_plot_lines(self):
'''
Test the capability of plotting along xz/yz orientation and use line plots based on drill hole id
'''
output_file = os.path.join(self.out_dir, 'location_plot3.png')
try:
_ = gs.location_plot(self.data,
var='Phi',
orient='yz',
aspect=5,
plot_collar=True,
output_file=output_file)
except Exception as ex:
self.fail(
'Unable to test the location plot along yz/xz orientation \n{}'
.format(str(ex)))
self.assertEqual(os.path.isfile(output_file), True)
import pygeostat as gs
# load data
data_file = gs.ExampleData('point3d_ind_mv')
gs.location_plot(data_file, var='Phi', orient='yz', aspect=5, plot_collar=True)
import pygeostat as gs
data_file = gs.ExampleData('point3d_ind_mv')
gs.location_plot(data_file)
import pygeostat as gs
data_file = gs.ExampleData('point3d_ind_mv')
gs.location_plot(data_file, var='Phi')
import pygeostat as gs
# load data
data_file = gs.ExampleData('point3d_ind_mv')
gs.location_plot(data_file,
var='Lithofacies',
orient='yz',
aspect=5,
plot_collar=True)
