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])
