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