Increase codecoverage

PlasmaControl · Oct 20, 2024 · b9e867e · b9e867e
1 parent b229726
commit b9e867e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_integrals.py b/tests/test_integrals.py
@@ -1776,6 +1776,7 @@ def test_bounce2d_checks(self):
 
         # 10. Plotting
         fig, ax = bounce.plot(l, pitch_inv[l], include_legend=False, show=False)
+        _, _ = bounce.plot_theta(l, show=False)
 
         # make sure tests pass when spline=True
         b = Bounce2D(
@@ -1788,6 +1789,7 @@ def test_bounce2d_checks(self):
             spline=True,
         )
         b.check_points(b.points(pitch_inv), pitch_inv, plot=False)
+        _, _ = bounce.plot(l, pitch_inv[l], show=False)
 
         return fig