Coverage for python/gsfit/plotting/two_d_grid.py: 0%

1import matplotlib.axes 

2 

3from ..gsfit import Gsfit 

4 

5 

6def plot( 

7 gsfit_controller: Gsfit, 

8 ax: matplotlib.axes.Axes, 

9) -> None: 

10 plasma = gsfit_controller.plasma 

11 

12 r = plasma.get_array1(["grid", "r"]) 

13 z = plasma.get_array1(["grid", "z"]) 

14 n_r = len(r) 

15 n_z = len(z) 

16 

17 for i_z in range(n_z): 

18 ax.plot([r[0], r[-1]], [z[i_z], z[i_z]], color="black", linestyle="solid", linewidth=0.5) 

19 for i_r in range(n_r): 

20 ax.plot([r[i_r], r[i_r]], [z[0], z[-1]], color="black", linestyle="solid", linewidth=0.5) 

21 

22 flat_r = plasma.get_array1(["grid", "flat", "r"]) 

23 flat_z = plasma.get_array1(["grid", "flat", "z"]) 

24 ax.plot(flat_r, flat_z, color="black", linestyle="none", marker="o", markersize=0.5)