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

1import matplotlib.axes 

2import numpy as np 

3from gsfit_rs.imas import equilibrium_paths as ep 

4 

5from ..gsfit import Gsfit 

6 

7 

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

9 plasma = gsfit_controller.plasma 

10 

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

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

13 

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

15 p_prime = plasma.get(ep.time_slice[i_time].profiles_1d.dpressure_dpsi) 

16 

17 ax.plot(psi_n, p_prime, color=color, linestyle=linestyle, label="GSFit", linewidth=1.0)