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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
import pytest
import numpy as np
from dynasor.post_processing import Weights
@pytest.fixture
def weights_coh():
weights_coh = dict()
weights_coh['A'] = 1.40
weights_coh['B'] = 0.98
weights_coh['some-group'] = 12.99
return weights_coh
@pytest.fixture
def weights_incoh():
weights_incoh = dict()
weights_incoh['A'] = 23.2
weights_incoh['B'] = 11.0
weights_incoh['some-group'] = 45.8
return weights_incoh
def test_init_weights(weights_coh, weights_incoh):
# without incoherent
weights = Weights(weights_coh)
assert weights.supports_currents
assert not weights.supports_incoherent
# with incoherent
weights = Weights(weights_coh, weights_incoh)
assert weights.supports_currents
assert weights.supports_incoherent
# without currents
weights = Weights(weights_coh, weights_incoh, supports_currents=False)
assert not weights.supports_currents
assert weights.supports_incoherent
def test_get_weights(weights_coh, weights_incoh):
weights = Weights(weights_coh, weights_incoh)
assert weights.supports_currents
assert weights.supports_incoherent
# check coherent weights
for key, val in weights_coh.items():
assert np.isclose(val, weights.get_weight_coh(key))
# check incoherent weights
for key, val in weights_incoh.items():
assert np.isclose(val, weights.get_weight_incoh(key))
|
