def test_convert_units(self):
"""
Should convert header units to plot units.
"""
fig = plt.figure()
ax = fig.add_subplot(111)
splt = SEGYPlotManager(ax, self.segy)
# should correctly perform unit conversions for distance
splt.DISTANCE_UNIT = 'km'
self.assertEqual(splt._convert_units('offset', [1000]), [1])
# should correctly perform unit conversions for time
splt.TIME_UNIT = 's'
self.assertEqual(splt._convert_units('delay', [1000]), [1])
def test_get_units(self):
"""
Should return (segy units, plot units) or None.
"""
fig = plt.figure()
ax = fig.add_subplot(111)
splt = SEGYPlotManager(ax, self.segy)
splt.DISTANCE_UNIT = 'distance_unit_marker'
splt.TIME_UNIT = 'time_unit_marker'
for key in TRACE_HEADER_KEYS:
if key in splt.SEGY_TIME_UNITS:
# should return TIME_UNIT for a time attribute
self.assertEqual(splt._get_units(key)[1],
'time_unit_marker')
elif key in splt.SEGY_DISTANCE_UNITS:
# should return DISTANCE_UNIT for a distance attribute
self.assertEqual(splt._get_units(key)[1],
'distance_unit_marker')
else:
# should return None values are unitless
self.assertEqual(splt._get_units(key), None)
