mistake in guess

n-claes · Dec 17, 2024 · 8a86359 · 8a86359
1 parent dbb986b
commit 8a86359
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 16 deletions.
diff --git a/tests/pylbo_tests/test_gimli.py b/tests/pylbo_tests/test_gimli.py
@@ -2,30 +2,61 @@
 import sympy as sp
 import filecmp
 
+
 def test_variables():
     obj = gimli.Variables()
     keychain = obj.__dict__.keys()
     for key in [
-        'x', 'y', 'z',
-        'rho0', 'T0', 'B0sq','k2', 'k3',
-        'rhoc', 'Tc', 'B2c', 'B3c', 'v2c', 'v3c', 'pc',
-        'p1', 'p2', 'p3', 'p4', 'p5', 'p6', 'p7', 'p8',
-        'alpha', 'beta', 'delta', 'theta', 'tau', 'lamda', 'nu',
-        'r0', 'rc', 'rj', 'Bth0', 'Bz0', 'V', 'j0', 'g'
+        "x",
+        "y",
+        "z",
+        "rho0",
+        "T0",
+        "B0sq",
+        "k2",
+        "k3",
+        "rhoc",
+        "Tc",
+        "B2c",
+        "B3c",
+        "v2c",
+        "v3c",
+        "pc",
+        "p1",
+        "p2",
+        "p3",
+        "p4",
+        "p5",
+        "p6",
+        "p7",
+        "p8",
+        "alpha",
+        "beta",
+        "delta",
+        "theta",
+        "tau",
+        "lamda",
+        "nu",
+        "r0",
+        "rc",
+        "rj",
+        "Bth0",
+        "Bz0",
+        "V",
+        "j0",
+        "g",
     ]:
         assert key in keychain
         assert isinstance(obj.__dict__[key], sp.Symbol)
-    assert 'fkey' in keychain
-    assert isinstance(obj.__dict__['fkey'], dict)
+    assert "fkey" in keychain
+    assert isinstance(obj.__dict__["fkey"], dict)
 
 
 def test_equilibrium():
     var = gimli.Variables()
     obj = gimli.Equilibrium(var, var.rhoc, 0, 0, var.Tc)
     keychain = obj.__dict__.keys()
-    for key in [
-        'variables', 'rho0', 'v02', 'v03', 'T0', 'B02', 'B03', '_dict_phys'
-    ]:
+    for key in ["variables", "rho0", "v02", "v03", "T0", "B02", "B03", "_dict_phys"]:
         assert key in keychain
     assert isinstance(obj.get_physics(), dict)
     assert isinstance(obj.get_dependencies(), dict)
@@ -45,24 +76,28 @@ def test_legolas_userfile(tmpdir, mod_usr):
         },
         "equilibrium_type": "user_defined",
         "physics_type": "mhd",
-        "logging_level": 1
+        "logging_level": 1,
     }
     var = gimli.Variables()
     eq = gimli.Equilibrium(var, var.rhoc, 0, 0, var.Tc)
     obj = gimli.Legolas(eq, config)
     obj.user_module(loc=tmpdir)
-    assert filecmp.cmp((tmpdir / 'smod_user_defined.f08').resolve(), mod_usr, shallow=False)
+    assert filecmp.cmp(
+        (tmpdir / "smod_user_defined.f08").resolve(), mod_usr, shallow=False
+    )
 
 
 def test_amrvac_preparation(tmpdir, datv211_harris, vacv211_harris):
     config = {
         "datfile": datv211_harris,
         "physics_type": "mhd",
-        "ev_guess": [0.01636j, 0.4323-6.391e-5*1j, -0.4323-6.391e-5*1j],
+        "ev_guess": [0.01636j, 1.397e-2 - 2.843e-4 * 1j, -1.397e-2 - 2.843e-4 * 1j],
         "ev_time": 0,
         "percentage": 0.01,
-        "quantity": "B02"
+        "quantity": "B02",
     }
     amrvac = gimli.Amrvac(config)
     amrvac.prepare_legolas_data(loc=tmpdir)
-    assert filecmp.cmp((tmpdir / 'v2.1.1_harris.ldat').resolve(), vacv211_harris, shallow=False)
+    assert filecmp.cmp(
+        (tmpdir / "v2.1.1_harris.ldat").resolve(), vacv211_harris, shallow=False
+    )
diff --git a/tests/pylbo_tests/utility_files/v2.1.1_harris.ldat b/tests/pylbo_tests/utility_files/v2.1.1_harris.ldat