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

1import matplotlib.axes 

2import numpy as np 

3 

4from ..gsfit import Gsfit 

5 

6 

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

8 plasma = gsfit_controller.plasma 

9 

10 time = plasma.get_array1(["time"]) 

11 i_time = np.argmin(np.abs(time - time_desired)) 

12 

13 psi_n = plasma.get_array1(["profiles", "psi_n"]) 

14 p_prime = plasma.get_array2(["profiles", "p_prime"])[i_time, :] 

15 

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