test: updates old test to use snapshot tests

This commit is contained in:
Deepak Mallubhotla 2023-07-23 15:30:30 -05:00
parent 857391425f
commit e9bb62c0a0
Signed by: deepak
GPG Key ID: BEBAEBF28083E022
2 changed files with 36 additions and 3 deletions

View File

@ -0,0 +1,31 @@
# serializer version: 1
# name: test_random_count_multiple_dipole_fixed_or_fixed_mag_model_get_n_dipoles
list([
list([
list([
10.0,
0.0,
6.123233995736766e-16,
-2.3960852996076447,
4.232462337639554,
2.2616924238635443,
39.399008444891905,
]),
]),
])
# ---
# name: test_random_count_multiple_dipole_fixed_or_fixed_mag_model_get_n_dipoles.1
list([
list([
list([
10.0,
0.0,
6.123233995736766e-16,
-2.3960852996076447,
4.232462337639554,
2.2616924238635443,
39.399008444891905,
]),
]),
])
# ---

View File

@ -115,7 +115,7 @@ def test_random_count_multiple_dipole_fixed_mag_model_get_dipoles_invariant():
)
def test_random_count_multiple_dipole_fixed_or_fixed_mag_model_get_n_dipoles():
def test_random_count_multiple_dipole_fixed_or_fixed_mag_model_get_n_dipoles(snapshot):
# TODO: this test is a bit garbage just calls things without testing.
x_min = -10
x_max = 10
@ -145,7 +145,9 @@ def test_random_count_multiple_dipole_fixed_or_fixed_mag_model_get_n_dipoles():
)
model.rng = numpy.random.default_rng(1234)
model.get_monte_carlo_dipole_inputs(1, max_frequency)
model.get_monte_carlo_dipole_inputs(
actual_inputs = model.get_monte_carlo_dipole_inputs(1, max_frequency)
assert actual_inputs.tolist() == snapshot
actual_monte_carlo_inputs = model.get_monte_carlo_dipole_inputs(
1, max_frequency, numpy.random.default_rng(1234)
)
assert actual_monte_carlo_inputs.tolist() == snapshot