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)