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))