Trying to identify error in pytest

petrobras · Jan 15, 2025 · bcc68b0 · bcc68b0
1 parent f271a31
commit bcc68b0
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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)