예제 #1
0
 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))