Пример #1
0
def test_min_max():
    vec1 = np.arange(2, 6)
    vec2 = np.arange(-4, 5, 2)
    vec3 = np.arange(-1, 1)
    scalar = 0.5

    grid = TensorGrid(vec1, vec2, vec3, as_midp=False)
    assert all_equal(grid.min(), (2, -4, -1))
    assert all_equal(grid.max(), (5, 4, 0))

    grid = TensorGrid(vec1, scalar, vec2, scalar, as_midp=False)
    assert all_equal(grid.min(), (2, 0.5, -4, 0.5))
    assert all_equal(grid.max(), (5, 0.5, 4, 0.5))

    grid = TensorGrid(vec1, vec2, vec3, as_midp=True)
    assert all_equal(grid.min(), (1.5, -5, -1.5))
    assert all_equal(grid.max(), (5.5, 5, 0.5))

    grid = TensorGrid(vec1, scalar, vec2, scalar, as_midp=True)
    assert all_equal(grid.min(), (1.5, 0.5, -5, 0.5))
    assert all_equal(grid.max(), (5.5, 0.5, 5, 0.5))