Deepak Mallubhotla
d89d3585da
All checks were successful
gitea-physics/pdme/pipeline/head This commit looks good
129 lines
3.7 KiB
Python
129 lines
3.7 KiB
Python
import numpy
|
|
import pdme.measurement
|
|
|
|
|
|
def test_static_dipole():
|
|
d1 = pdme.measurement.OscillatingDipole((1, 2, 3), (4, 5, 6), 7)
|
|
d2 = pdme.measurement.OscillatingDipole((2, 5, 3), (4, -5, -6), 2)
|
|
dipoles = pdme.measurement.OscillatingDipoleArrangement([d1, d2])
|
|
|
|
dot_position1 = (-1, -1, -1)
|
|
dot_frequency1 = 11
|
|
expected_v1 = 0.00001421963287022476
|
|
expected_v2 = 0.00001107180225755457
|
|
|
|
numpy.testing.assert_allclose(
|
|
d1.s_at_position(dot_position1, dot_frequency1),
|
|
expected_v1,
|
|
err_msg="Voltage at dot isn't as expected.",
|
|
)
|
|
|
|
dot_measurements = dipoles.get_dot_measurements([(dot_position1, dot_frequency1)])
|
|
assert len(dot_measurements) == 1, "Should have only had one dot measurement."
|
|
measurement = dot_measurements[0]
|
|
numpy.testing.assert_allclose(
|
|
measurement.r,
|
|
dot_position1,
|
|
err_msg="Dot position should have been passed over",
|
|
)
|
|
numpy.testing.assert_allclose(
|
|
measurement.f,
|
|
dot_frequency1,
|
|
err_msg="Dot frequency should have been passed over",
|
|
)
|
|
numpy.testing.assert_allclose(
|
|
measurement.v,
|
|
expected_v1 + expected_v2,
|
|
err_msg="Voltage at dot isn't as expected.",
|
|
)
|
|
|
|
|
|
def test_dipole_dot_pair():
|
|
d1 = pdme.measurement.OscillatingDipole((1, 2, 3), (4, 5, 6), 7)
|
|
dipoles = pdme.measurement.OscillatingDipoleArrangement([d1])
|
|
|
|
dot_position1 = (-1, -1, -1)
|
|
dot_position2 = (1, 2, 3)
|
|
dot_frequency = 8
|
|
expected_sij = 0.000083328037100902801698
|
|
|
|
numpy.testing.assert_allclose(
|
|
d1.s_for_dot_pair(dot_position1, dot_position2, dot_frequency),
|
|
expected_sij,
|
|
err_msg="Sij for dot pair isn't as expected.",
|
|
)
|
|
numpy.testing.assert_allclose(
|
|
[
|
|
m.v
|
|
for m in dipoles.get_dot_pair_measurements(
|
|
[(dot_position1, dot_position2, dot_frequency)]
|
|
)
|
|
],
|
|
[expected_sij],
|
|
err_msg="Sij for dot pair isn't as expected via dipole.",
|
|
)
|
|
|
|
|
|
def test_range_pairs():
|
|
d1 = pdme.measurement.OscillatingDipole((1, 2, 3), (4, 5, 6), 7)
|
|
dipoles = pdme.measurement.OscillatingDipoleArrangement([d1])
|
|
|
|
dot_position1 = (-1, -1, -1)
|
|
dot_position2 = (1, 2, 3)
|
|
dot_frequency = 8
|
|
expected_sij = 0.000083328037100902801698
|
|
|
|
actuals = dipoles.get_percent_range_dot_pair_measurements(
|
|
[(dot_position1, dot_position2, dot_frequency)], 0.5, 1.5
|
|
)
|
|
assert len(actuals) == 1, "should have only been one pair"
|
|
actual = actuals[0]
|
|
numpy.testing.assert_allclose(
|
|
[actual.v_low, actual.v_high],
|
|
expected_sij * numpy.array([0.5, 1.5]),
|
|
err_msg="Sij for dot pair isn't as expected via dipole with range.",
|
|
)
|
|
|
|
|
|
def test_range_dipole_measurements():
|
|
d1 = pdme.measurement.OscillatingDipole((1, 2, 3), (4, 5, 6), 7)
|
|
d2 = pdme.measurement.OscillatingDipole((2, 5, 3), (4, -5, -6), 2)
|
|
dipoles = pdme.measurement.OscillatingDipoleArrangement([d1, d2])
|
|
|
|
dot_position1 = (-1, -1, -1)
|
|
dot_frequency1 = 11
|
|
expected_v1 = 0.00001421963287022476
|
|
expected_v2 = 0.00001107180225755457
|
|
|
|
numpy.testing.assert_allclose(
|
|
d1.s_at_position(dot_position1, dot_frequency1),
|
|
expected_v1,
|
|
err_msg="Voltage at dot isn't as expected.",
|
|
)
|
|
|
|
range_dot_measurements = dipoles.get_percent_range_dot_measurements(
|
|
[(dot_position1, dot_frequency1)], 0.5, 1.5
|
|
)
|
|
assert len(range_dot_measurements) == 1, "Should have only had one dot measurement."
|
|
range_measurement = range_dot_measurements[0]
|
|
numpy.testing.assert_allclose(
|
|
range_measurement.r,
|
|
dot_position1,
|
|
err_msg="Dot position should have been passed over",
|
|
)
|
|
numpy.testing.assert_allclose(
|
|
range_measurement.f,
|
|
dot_frequency1,
|
|
err_msg="Dot frequency should have been passed over",
|
|
)
|
|
numpy.testing.assert_allclose(
|
|
range_measurement.v_low,
|
|
(expected_v1 + expected_v2) / 2,
|
|
err_msg="Lower voltage at dot isn't as expected.",
|
|
)
|
|
numpy.testing.assert_allclose(
|
|
range_measurement.v_high,
|
|
(expected_v1 + expected_v2) * 3 / 2,
|
|
err_msg="Lower oltage at dot isn't as expected.",
|
|
)
|