Fix failing docstring test for NaN values in "has_cp_curve"

birgits · birgits · commit 3a6ed88234de · 2024-02-01T10:20:25.000-08:00
diff --git a/windpowerlib/data.py b/windpowerlib/data.py
@@ -92,10 +92,10 @@ def get_turbine_types(turbine_library="local", print_out=True, filter_=True):
             + "but must be 'local' or 'oedb'."
         )
     if filter_:
-        cp_curves_df = df.loc[df["has_cp_curve"]][
+        cp_curves_df = df.loc[df["has_cp_curve"].fillna(False)][
             ["manufacturer", "turbine_type", "has_cp_curve"]
         ]
-        p_curves_df = df.loc[df["has_power_curve"]][
+        p_curves_df = df.loc[df["has_power_curve"].fillna(False)][
             ["manufacturer", "turbine_type", "has_power_curve"]
         ]
         curves_df = pd.merge(

Original file line number	Diff line number	Diff line change
`@@ -92,10 +92,10 @@ def get_turbine_types(turbine_library="local", print_out=True, filter_=True):`
`92`	`92`	`+ "but must be 'local' or 'oedb'."`
`93`	`93`	`)`
`94`	`94`	`if filter_:`
`95`		`- cp_curves_df = df.loc[df["has_cp_curve"]][`
	`95`	`+ cp_curves_df = df.loc[df["has_cp_curve"].fillna(False)][`
`96`	`96`	`["manufacturer", "turbine_type", "has_cp_curve"]`
`97`	`97`	`]`
`98`		`- p_curves_df = df.loc[df["has_power_curve"]][`
	`98`	`+ p_curves_df = df.loc[df["has_power_curve"].fillna(False)][`
`99`	`99`	`["manufacturer", "turbine_type", "has_power_curve"]`
`100`	`100`	`]`
`101`	`101`	`curves_df = pd.merge(`