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)