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
30
31
32
33
34
35
36
37
38
39
|
#! /usr/bin/env python
import openturns as ot
import numpy as np
ot.TESTPREAMBLE()
sample = ot.Normal(3).getSample(1000)
tree = ot.KDTree(sample)
print("tree=", tree)
test = ot.Normal(3).getSample(100)
sample_np = np.array(sample)
test_np = np.array(test)
def nearest_debug(x):
global sample_np
a = np.sum(np.square(sample_np - x), axis=1)
return int(np.argmin(a))
def nearest_debug_indices(x):
global sample_np
a = np.sum(np.square(sample_np - x), axis=1)
return a.argsort()
neighbourIndices = tree.query(test)
neighbourIndices_np = [nearest_debug(x) for x in test]
assert neighbourIndices == neighbourIndices_np, "Errors found in query"
for x in test:
assert np.any(
nearest_debug_indices(x)[:10] == tree.queryK(x, 10, True)
), "Errors found in queryK"
|
