Example #1
0
def test_swapaxis_yflip(default_surface, yflip, expected_result):
    default_surface["yflip"] = yflip
    surface = RegularSurface(**default_surface)

    surface.swapaxes()

    assert surface.yflip == expected_result
Example #2
0
def test_swapaxis_xinc_yinc(default_surface):
    default_surface["yinc"] = 2.0
    surface = RegularSurface(**default_surface)

    surface.swapaxes()

    assert (surface.xinc, surface.yinc) == (2, 1)
Example #3
0
def test_swapaxis_rotation(rotation, expected_rotation, default_surface):
    default_surface["rotation"] = rotation
    surface = RegularSurface(**default_surface)

    surface.swapaxes()

    assert surface.rotation == expected_rotation
Example #4
0
def test_swapaxis_ncol_nrow(default_surface):
    default_surface["nrow"] = 3
    default_surface["values"] = [1] * 6
    surface = RegularSurface(**default_surface)

    surface.swapaxes()

    assert (surface.nrow, surface.ncol) == (2, 3)
Example #5
0
def test_swapaxis_ilines(default_surface):
    surface = RegularSurface(**default_surface)

    assert surface.ilines.tolist() == [1, 2]

    surface.swapaxes()

    assert surface.ilines.tolist() == [1, 2]
Example #6
0
def test_swapaxis(default_surface):
    surface = RegularSurface(**default_surface)

    assert surface.values.flatten().tolist() == [1, 2, 3, 4]

    surface.swapaxes()

    assert surface.values.flatten().tolist() == [1.0, 3.0, 2.0, 4.0]