From e0e8411a81094b850b203c1d1000f8dfaf914763 Mon Sep 17 00:00:00 2001
From: Pauli Virtanen <pauli.t.virtanen@jyu.fi>
Date: Mon, 29 Aug 2022 14:46:55 +0300
Subject: [PATCH] tests: test currents

---
 tests/test_solver.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/test_solver.py b/tests/test_solver.py
index 7396d8d..37643cb 100644
--- a/tests/test_solver.py
+++ b/tests/test_solver.py
@@ -55,7 +55,7 @@ def test_example_sns_1d():
     assert_allclose(rF.squeeze(), F, rtol=0, atol=5e-3 * abs(F).max())
 
 
-@pytest.mark.skip  # TODO
+@pytest.mark.slow
 @pytest.mark.skipif(usadel1 is None, reason="usadel1 not installed")
 def test_example_sns_1d_J_usadel1():
     Lx = 10
@@ -66,7 +66,8 @@ def test_example_sns_1d_J_usadel1():
     T = 0.2
     Delta, I0, _, success = sol.self_consistency(T=T, T_c0=0)
 
-    I0 = I0[0, :-1, 0, 0]
+    tau3 = np.diag([1, 1, -1, -1])
+    I0 = (I0[:-1, 0, 0] @ tau3).trace(axis1=-2, axis2=-1)
 
     I0_mean = I0.real.mean()
 
-- 
GitLab