def test_swapaxis_xinc_yinc():
cube = Cube(
xori=0.0,
yori=0.0,
zori=0.0,
ncol=2,
nrow=3,
nlay=2,
xinc=1.0,
yinc=2.0,
zinc=1.0,
yflip=1,
)
cube.swapaxes()
assert (cube.xinc, cube.yinc) == (2, 1)
def test_cube_swapaxes():
"""Import a cube, do axes swapping back and forth"""
logger.info("Import SEGY format via SEGYIO")
incube = Cube(SFILE4)
logger.info(incube)
val1 = incube.values.copy()
incube.swapaxes()
logger.info(incube)
incube.swapaxes()
val2 = incube.values.copy()
logger.info(incube)
np.testing.assert_array_equal(val1, val2)
def test_swapaxis_ncol_nrow():
cube = Cube(
xori=0.0,
yori=0.0,
zori=0.0,
ncol=2,
nrow=3,
nlay=2,
xinc=1.0,
yinc=1.0,
zinc=1.0,
yflip=1,
)
cube.swapaxes()
assert (cube.nrow, cube.ncol) == (2, 3)
def test_swapaxis_ilines():
cube = Cube(
xori=0.0,
yori=0.0,
zori=0.0,
ncol=2,
nrow=2,
nlay=2,
xinc=1.0,
yinc=1.0,
zinc=1.0,
yflip=1,
values=[1, 2, 3, 4, 5, 6, 7, 8],
)
assert cube.ilines.tolist() == [1, 2]
cube.swapaxes()
assert cube.ilines.tolist() == [1, 2]
def test_swapaxis_rotation(rotation, expected_rotation):
cube = Cube(
xori=0.0,
yori=0.0,
zori=0.0,
ncol=2,
nrow=2,
nlay=2,
xinc=1.0,
yinc=1.0,
zinc=1.0,
yflip=1,
rotation=rotation,
values=[1, 2, 3, 4, 5, 6, 7, 8],
)
cube.swapaxes()
assert cube.rotation == expected_rotation
def test_swapaxis_traceidcodes():
cube = Cube(
xori=0.0,
yori=0.0,
zori=0.0,
ncol=2,
nrow=2,
nlay=2,
xinc=1.0,
yinc=1.0,
zinc=1.0,
yflip=1,
values=[1, 2, 3, 4, 5, 6, 7, 8],
)
assert cube.traceidcodes.flatten().tolist() == [1, 1, 1, 1]
cube.traceidcodes = [1, 2, 3, 4]
cube.swapaxes()
assert cube.traceidcodes.flatten().tolist() == [1, 3, 2, 4]
def test_swapaxis():
cube = Cube(
xori=0.0,
yori=0.0,
zori=0.0,
ncol=2,
nrow=2,
nlay=2,
xinc=1.0,
yinc=1.0,
zinc=1.0,
yflip=1,
values=[1, 2, 3, 4, 5, 6, 7, 8],
)
assert cube.values.flatten().tolist() == [1, 2, 3, 4, 5, 6, 7, 8]
cube.swapaxes()
assert cube.values.flatten().tolist() == [1.0, 2.0, 5.0, 6.0, 3.0, 4.0, 7.0, 8.0]
def test_cube_swapaxes():
"""Import a cube, do axes swapping back and forth"""
logger.info("Import SEGY format via SEGYIO")
incube = Cube(SFILE4)
logger.info(incube)
val1 = incube.values.copy()
incube.swapaxes()
logger.info(incube)
incube.swapaxes()
val2 = incube.values.copy()
logger.info(incube)
diff = val1 - val2
tsetup.assert_almostequal(diff.mean(), 0.0, 0.000001)
tsetup.assert_almostequal(diff.std(), 0.0, 0.000001)
assert incube.ilines.size == incube.ncol
