def test_constant_dist(fn, exponent):
[xarr, yarr], [x, y] = noise_elements(fn, 2)
constant = 1.5
if exponent == float('inf'):
factor = constant
else:
factor = constant**(1 / exponent)
true_dist = factor * np.linalg.norm(xarr - yarr, ord=exponent)
w_const = NumpyFnConstWeighting(constant, exponent=exponent)
assert almost_equal(w_const.dist(x, y), true_dist)
# With free function
w_const_dist = npy_weighted_dist(constant, exponent=exponent)
assert almost_equal(w_const_dist(x, y), true_dist)
def test_const_dist_using_inner(fn):
[xarr, yarr], [x, y] = noise_elements(fn, 2)
constant = 1.5
w = NumpyFnConstWeighting(constant)
true_dist = np.sqrt(constant) * np.linalg.norm(xarr - yarr)
# Using 3 places (single precision default) since the result is always
# double even if the underlying computation was only single precision
assert almost_equal(w.dist(x, y), true_dist, places=3)
# Only possible for exponent=2
with pytest.raises(ValueError):
NumpyFnConstWeighting(constant, exponent=1, dist_using_inner=True)
# With free function
w_dist = npy_weighted_dist(constant, use_inner=True)
assert almost_equal(w_dist(x, y), true_dist, places=3)
