def test_build_header_database(self):
"""
Should connect to a database for looking up header attributes.
"""
fig = plt.figure()
ax = fig.add_subplot(111)
# connecting to a database in memory should not create a file
splt = SEGYPlotManager(ax, self.segy, trace_header_database=':memory:')
self.assertFalse(os.path.isfile(':memory:'))
# connecting to a filename should create a file if it does not exist
filename = 'temp.test_build_header_database.sqlite'
if os.path.isfile(filename):
os.remove(filename)
# should not create database if no pickdb is given
splt = SEGYPlotManager(ax, self.segy, trace_header_database=filename)
self.assertFalse(os.path.isfile(filename))
# should create database if pickdb is given
splt = SEGYPlotManager(ax, self.segy, pickdb=self.pickdb,
trace_header_database=filename)
self.assertTrue(os.path.isfile(filename))
# clean up
os.remove(filename)
# create plot manager in memory with pickdb
splt = SEGYPlotManager(ax, self.segy, pickdb=self.pickdb)
# should be able to get primary fields from the picks trace table
pick_keys = splt.pickdb._get_primary_fields(splt.pickdb.TRACE_TABLE)
self.assertTrue(len(pick_keys) > 0)
# pick primary keys should be in the alias dictionary
# or be a header attribute
header = self.segy.traces[0].header
for k in pick_keys:
self.assertTrue(k in splt.SEGY_HEADER_ALIASES \
or hasattr(header, k))
# should add header fields for primary keys in the pick database
sql = 'SELECT * FROM %s' %splt.sdb.TRACE_TABLE
data = splt.sdb.execute(sql)
for key in pick_keys:
_key = splt._get_header_alias(key)
for i,row in enumerate(data):
segy_value = splt.get_header_value(
self.segy.traces[i].header, key, convert_units=False)
db_value = row[_key]
self.assertEqual(segy_value, db_value)
def test_get_abscissa(self):
"""
Should return a list of abcissa values.
"""
fig = plt.figure()
ax = fig.add_subplot(111)
splt = SEGYPlotManager(ax, self.segy, pickdb=self.pickdb)
# just get abcissa for a subset of the traces
idx = [0, 100, 103, 104, 500, 550]
values = []
for i in idx:
ensemble = self.segy.traces[i].header.ensemble_number
trace = \
self.segy.traces[i].header.trace_number_within_the_ensemble
values.append((ensemble, trace))
# should return a list of values for plot x-axis
x = splt._get_abscissa(['ensemble', 'trace'], values)
_key = splt._get_header_alias(splt.ABSCISSA_KEY)
for j,i in enumerate(idx):
tr = self.segy.traces[i]
self.assertEqual(tr.header.__getattribute__(_key),
x[j])
