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

1import matplotlib.axes 

2import numpy as np 

3from gsfit_rs.imas import equilibrium_paths as ep 

4from scipy.constants import mu_0 

5 

6from ..gsfit import Gsfit 

7 

8 

9def plot(gsfit_controller: Gsfit, ax: matplotlib.axes.Axes, time_desired: float, color: str = "blue", linestyle: str = "solid") -> None: 

10 plasma = gsfit_controller.plasma 

11 

12 time = plasma.get(ep.time_slice[:].time) 

13 i_time = int(np.argmin(np.abs(time - time_desired))) 

14 

15 psi_n = plasma.get(ep.time_slice[0].profiles_1d.psi_norm) 

16 ff_prime = np.asarray(plasma.get(ep.time_slice[i_time].profiles_1d.f_df_dpsi)) / mu_0 

17 

18 ax.plot(psi_n, ff_prime, color=color, linestyle=linestyle, label="GSFit", linewidth=1.0)