From bcc68b0e9354fad64f904e123ede3698efeb9942 Mon Sep 17 00:00:00 2001 From: jguarato Date: Wed, 15 Jan 2025 10:36:31 -0300 Subject: [PATCH] Trying to identify error in pytest --- ross/tests/test_stochastic_rotor_assembly.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/ross/tests/test_stochastic_rotor_assembly.py b/ross/tests/test_stochastic_rotor_assembly.py index 8aaffbc7..4b5eca80 100644 --- a/ross/tests/test_stochastic_rotor_assembly.py +++ b/ross/tests/test_stochastic_rotor_assembly.py @@ -300,7 +300,13 @@ def test_run_campbell(rotor1): assert results.wd.shape == (7, 11, 2) assert results.log_dec.shape == (7, 11, 2) - assert_allclose(results.wd, wd, atol=1e-6) + assert_allclose(results.wd[0, :, :], wd[0, :, :], atol=1e-6) + assert_allclose(results.wd[1, :, :], wd[1, :, :], atol=1e-6) + assert_allclose(results.wd[2, :, :], wd[2, :, :], atol=1e-6) + assert_allclose(results.wd[3, :, :], wd[3, :, :], atol=1e-6) + assert_allclose(results.wd[4, :, :], wd[4, :, :], atol=1e-6) + assert_allclose(results.wd[5, :, :], wd[5, :, :], atol=1e-6) + assert_allclose(results.wd[6, :, :], wd[6, :, :], atol=1e-6) assert_allclose(results.log_dec, log_dec, atol=1e-6)