def test_write_text_header(self): fname = self.filename.replace("small", "txt_hdr_wrt") shutil.copyfile(self.filename, fname) f = _segyio.open(fname, "r+") with self.assertRaises(ValueError): _segyio.write_textheader(f, 0, "") self.assertEqual(_segyio.read_textheader(f, 0), ACTUAL_TEXT_HEADER) _segyio.write_textheader(f, 0, "yolo" * 800) self.assertEqual(_segyio.read_textheader(f, 0), "yolo" * 800) _segyio.close(f)
def test_read_text_header(self): f = _segyio.open(self.filename, "r") self.assertEqual(_segyio.read_textheader(f, 0), ACTUAL_TEXT_HEADER) with self.assertRaises(Exception): _segyio.read_texthdr(None, 0) _segyio.close(f)
def test_write_text_header(self): with TestContext("write_text_header") as context: context.copy_file(self.filename) f = _segyio.open("small.sgy", "r+") with self.assertRaises(ValueError): _segyio.write_textheader(f, 0, "") self.assertEqual(_segyio.read_textheader(f, 0), self.ACTUAL_TEXT_HEADER) _segyio.write_textheader(f, 0, "yolo" * 800) textheader = _segyio.read_textheader(f, 0) textheader = textheader.decode('ascii') # Because in Python 3.5 bytes are not comparable to strings self.assertEqual(textheader, "yolo" * 800) _segyio.close(f)
def test_write_text_header(self): with TestContext("write_text_header") as context: context.copy_file(self.filename) f = _segyio.open("small.sgy", "r+") _segyio.write_textheader(f, 0, "") textheader = _segyio.read_textheader(f, 0) textheader = textheader.decode('ascii') self.assertEqual(textheader, "" * 3200) _segyio.write_textheader(f, 0, "yolo" * 800) textheader = _segyio.read_textheader(f, 0) textheader = textheader.decode( 'ascii' ) # Because in Python 3.5 bytes are not comparable to strings self.assertEqual(textheader, "yolo" * 800) _segyio.close(f)
def __getitem__(inner, index): if index > self.ext_headers: raise IndexError("Textual header %d not in file" % index) return _segyio.read_textheader(self.xfd, index)
def __getitem__(self, index): if index > self.outer.ext_headers: raise IndexError("Textual header {} not in file".format(index)) return _segyio.read_textheader(self.outer.xfd, index)