Test shared profile geometry

Author: nicamo

freegsnke/tests/test_jtor_update.py

@@ -76,3 +76,29 @@ def test_profiles_BetapIp(create_machine):
         and hasattr(profiles, "opt")
         and hasattr(profiles, "jtor")
     ), "The profiles object does not have the xpt, opt and jtor attributes"
+
+
+def test_profile_geometry_is_shared_and_fallback_indices_are_lazy(create_machine):
+    eq, _ = create_machine
+    profiles = jtor.ConstrainPaxisIp(eq, 8.1e3, 6.2e5, 0.5)
+    profiles.inputs = []
+    copied_profiles = jtor.Jtor_universal.copy(profiles)
+    handler = eq.limiter_handler
+
+    for name in (
+        "dR_dZ",
+        "R0Z0",
+        "eqRidx",
+        "eqZidx",
+        "mask_inside_limiter",
+        "mask_outside_limiter",
+        "limiter_mask_out",
+    ):
+        assert getattr(profiles, name) is getattr(handler, name)
+        assert getattr(copied_profiles, name) is getattr(handler, name)
+
+    assert handler._idx_grid_points is None
+    assert handler.idx_grid_points.shape == (eq.nx * eq.ny, 2)
+    expected_indices = np.indices((eq.nx, eq.ny)).reshape(2, -1).T
+    assert np.array_equal(handler.idx_grid_points, expected_indices)
+    assert handler.idx_grid_points is handler._idx_grid_points