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
|
# fmt: off
import numpy as np
from ase.dft import get_distribution_moment
def test_distmom():
precision = 1E-8
x = np.linspace(-50., 50., 1000)
y = np.exp(-x**2 / 2.)
area, center, mom2 = get_distribution_moment(x, y, (0, 1, 2))
assert sum((abs(area - np.sqrt(2. * np.pi)),
abs(center), abs(mom2 - 1.))) < precision
x = np.linspace(-1., 1., 100000)
for order in range(9):
y = x**order
area = get_distribution_moment(x, y)
assert abs(area - (1. - (-1.)**(order + 1)) / (order + 1.)) < precision
x = np.linspace(-50., 50., 100)
y = np.exp(-2. * (x - 7.)**2 / 10.) + np.exp(-2. * (x + 5.)**2 / 10.)
center = get_distribution_moment(x, y, 1)
assert abs(center - 1.) < precision
|
