def test_equals(as_midp):
vec1 = np.arange(2, 6)
vec2 = np.arange(-4, 5, 2)
grid1 = TensorGrid(vec1, as_midp=as_midp)
grid1_diff_midp = TensorGrid(vec1, as_midp=not as_midp)
grid2 = TensorGrid(vec1, vec2, as_midp=as_midp)
grid2_again = TensorGrid(vec1, vec2, as_midp=as_midp)
grid2_rev = TensorGrid(vec2, vec1, as_midp=as_midp)
assert grid1 == grid1
assert not grid1 != grid1
assert grid1 != grid1_diff_midp
assert grid2 == grid2
assert not grid2 != grid2
assert grid2 == grid2_again
assert not grid1 == grid2
assert not grid2 == grid2_rev
assert not grid2 == (vec1, vec2)
# Fuzzy check
grid1 = TensorGrid(vec1, vec2)
grid2 = TensorGrid(vec1 + (0.1, 0.05, 0, -0.1),
vec2 + (0.1, 0.05, 0, -0.1, -0.1))
assert grid1.approx_equals(grid2, tol=0.15)
assert grid2.approx_equals(grid1, tol=0.15)
grid2 = TensorGrid(vec1 + (0.11, 0.05, 0, -0.1),
vec2 + (0.1, 0.05, 0, -0.1, -0.1))
assert not grid1.approx_equals(grid2, tol=0.1)
grid2 = TensorGrid(vec1 + (0.1, 0.05, 0, -0.1),
vec2 + (0.1, 0.05, 0, -0.11, -0.1))
assert not grid1.approx_equals(grid2, tol=0.1)
def test_tensorgrid_equals():
vec1 = np.array([2, 3, 4, 5])
vec2 = np.array([-4, -2, 0, 2, 4])
grid1 = TensorGrid(vec1)
grid2 = TensorGrid(vec1, vec2)
grid2_again = TensorGrid(vec1, vec2)
grid2_rev = TensorGrid(vec2, vec1)
assert grid1 == grid1
assert not grid1 != grid1
assert grid2 == grid2
assert not grid2 != grid2
assert grid2 == grid2_again
assert not grid1 == grid2
assert not grid2 == grid2_rev
assert not grid2 == (vec1, vec2)
# Fuzzy check
grid1 = TensorGrid(vec1, vec2)
grid2 = TensorGrid(vec1 + (0.1, 0.05, 0, -0.1),
vec2 + (0.1, 0.05, 0, -0.1, -0.1))
assert grid1.approx_equals(grid1, atol=0.0)
assert grid1.approx_equals(grid2, atol=0.15)
assert grid2.approx_equals(grid1, atol=0.15)
grid2 = TensorGrid(vec1 + (0.11, 0.05, 0, -0.1),
vec2 + (0.1, 0.05, 0, -0.1, -0.1))
assert not grid1.approx_equals(grid2, atol=0.1)
grid2 = TensorGrid(vec1 + (0.1, 0.05, 0, -0.1),
vec2 + (0.1, 0.05, 0, -0.11, -0.1))
assert not grid1.approx_equals(grid2, atol=0.1)
def test_tensorgrid_equals():
vec1 = np.array([2, 3, 4, 5])
vec2 = np.array([-4, -2, 0, 2, 4])
grid1 = TensorGrid(vec1)
grid2 = TensorGrid(vec1, vec2)
grid2_again = TensorGrid(vec1, vec2)
grid2_rev = TensorGrid(vec2, vec1)
assert grid1 == grid1
assert not grid1 != grid1
assert grid2 == grid2
assert not grid2 != grid2
assert grid2 == grid2_again
assert not grid1 == grid2
assert not grid2 == grid2_rev
assert not grid2 == (vec1, vec2)
# Fuzzy check
grid1 = TensorGrid(vec1, vec2)
grid2 = TensorGrid(vec1 + (0.1, 0.05, 0, -0.1),
vec2 + (0.1, 0.05, 0, -0.1, -0.1))
assert grid1.approx_equals(grid1, atol=0.0)
assert grid1.approx_equals(grid2, atol=0.15)
assert grid2.approx_equals(grid1, atol=0.15)
grid2 = TensorGrid(vec1 + (0.11, 0.05, 0, -0.1),
vec2 + (0.1, 0.05, 0, -0.1, -0.1))
assert not grid1.approx_equals(grid2, atol=0.1)
grid2 = TensorGrid(vec1 + (0.1, 0.05, 0, -0.1),
vec2 + (0.1, 0.05, 0, -0.11, -0.1))
assert not grid1.approx_equals(grid2, atol=0.1)
