def test_get_time_array(self): """ Should return an array of time values for a single trace. """ fig = plt.figure() ax = fig.add_subplot(111) splt = SEGYPlotManager(ax, self.segy) tr = self.segy.traces[0] # should have npts values npts = splt.get_header_value(tr.header, 'npts') t = splt.get_time_array(tr.header) self.assertEqual(npts, len(t))