test: adds tests for phases and pairs
All checks were successful
gitea-physics/pdme/pipeline/head This commit looks good
All checks were successful
gitea-physics/pdme/pipeline/head This commit looks good
This commit is contained in:
parent
e61838d85f
commit
806b9b667f
@ -0,0 +1,13 @@
|
|||||||
|
# serializer version: 1
|
||||||
|
# name: test_fast_nonlocal_calc_multidipole_phase_snapshot
|
||||||
|
list([
|
||||||
|
list([
|
||||||
|
-0.0,
|
||||||
|
-0.0,
|
||||||
|
]),
|
||||||
|
list([
|
||||||
|
-0.0,
|
||||||
|
3.141592653589793,
|
||||||
|
]),
|
||||||
|
])
|
||||||
|
# ---
|
@ -53,3 +53,24 @@ def test_fast_nonlocal_frequency_check_multidipole():
|
|||||||
|
|
||||||
with pytest.raises(ValueError):
|
with pytest.raises(ValueError):
|
||||||
pdme.util.fast_nonlocal_spectrum.fast_s_nonlocal_dipoleses(dot_pairs, dipoles)
|
pdme.util.fast_nonlocal_spectrum.fast_s_nonlocal_dipoleses(dot_pairs, dipoles)
|
||||||
|
|
||||||
|
|
||||||
|
def test_fast_nonlocal_calc_multidipole_phase_snapshot(snapshot):
|
||||||
|
d1 = [1, 2, 3, 4, 5, 6, 7]
|
||||||
|
d2 = [1, 2, 3, 4, 5, 6, 8]
|
||||||
|
d3 = [2, 5, 3, 4, -5, -6, 2]
|
||||||
|
d4 = [-3, 2, 1, 4, 5, 6, 10]
|
||||||
|
|
||||||
|
dipoleses = numpy.array([[d1, d2], [d3, d4]])
|
||||||
|
|
||||||
|
dot_pairs = numpy.array(
|
||||||
|
[[[-1, -2, -3, 11], [-1, 2, 5, 11]], [[-1, -2, -3, 6], [2, 4, 6, 6]]]
|
||||||
|
)
|
||||||
|
|
||||||
|
# this is a bit silly but just set the logger to debug so that the coverage stats don't get affected by the debug statements.
|
||||||
|
pdme.util.fast_nonlocal_spectrum._logger.setLevel(logging.DEBUG)
|
||||||
|
|
||||||
|
actual_phases = pdme.util.fast_nonlocal_spectrum.signarg(
|
||||||
|
pdme.util.fast_nonlocal_spectrum.fast_s_nonlocal_dipoleses(dot_pairs, dipoleses)
|
||||||
|
)
|
||||||
|
assert actual_phases.tolist() == snapshot
|
||||||
|
Loading…
x
Reference in New Issue
Block a user