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

1import matplotlib.axes 

2import numpy as np 

3from scipy.constants import mu_0 

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_array1(["time"]) 

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

13 

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

15 ff_prime = plasma.get_array2(["profiles", "ff_prime"])[i_time, :] / mu_0 

16 

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