1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
import numpy
import pypathlib
def test_squared_distance():
path = pypathlib.Path([[0.0, 0.0], [1.0, 0.0], [0.9, 0.5], [1.0, 1.0], [0.0, 1.0]])
dist = path.squared_distance(
[[0.2, 0.1], [0.5, 0.5], [1.0, 0.5], [0.0, 1.1], [-0.1, 1.1], [1.0, 1.0]]
)
ref = numpy.array([0.01, 0.16, 1.0 / 104.0, 0.01, 0.02, 0.0])
assert numpy.all(numpy.abs(dist - ref) < 1.0e-12)
return
def test_one_point():
path = pypathlib.Path([[0.0, 0.0]])
dist = path.squared_distance(
[[0.2, 0.1], [0.5, 0.5], [1.0, 0.5], [0.0, 1.1], [-0.1, 1.1], [1.0, 1.0]]
)
ref = numpy.array([0.05, 0.5, 1.25, 1.21, 1.22, 2.0])
assert numpy.all(numpy.abs(dist - ref) < 1.0e-12)
return
if __name__ == "__main__":
test_squared_distance()
|
