BUG: discretize source when plotting drag curves if needed

rocketpy/plots/rocket_plots.py

@@ -134,10 +134,22 @@ def drag_curves(self):
         None
         """
 
-        x_power_drag_off = self.rocket.power_off_drag.x_array
-        y_power_drag_off = self.rocket.power_off_drag.y_array
-        x_power_drag_on = self.rocket.power_on_drag.x_array
-        y_power_drag_on = self.rocket.power_on_drag.y_array
+        try:
+            x_power_drag_on = self.rocket.power_on_drag.x_array
+            y_power_drag_on = self.rocket.power_on_drag.y_array
+        except AttributeError:
+            x_power_drag_on = np.linspace(0, 2, 50)
+            y_power_drag_on = np.array(
+                [self.rocket.power_on_drag.source(x) for x in x_power_drag_on]
+            )
+        try:
+            x_power_drag_off = self.rocket.power_off_drag.x_array
+            y_power_drag_off = self.rocket.power_off_drag.y_array
+        except AttributeError:
+            x_power_drag_off = np.linspace(0, 2, 50)
+            y_power_drag_off = np.array(
+                [self.rocket.power_off_drag.source(x) for x in x_power_drag_off]
+            )
 
         fig, ax = plt.subplots()
         ax.plot(x_power_drag_on, y_power_drag_on, label="Power on Drag")