Add to error raising

Author: SabineHaas

windpowerlib/wake_losses.py

@@ -87,9 +87,12 @@ def get_wind_efficiency_curve(curve_name='dena_mean'):
              p. 124
 
     """
+    possible_curve_names = ['dena_mean', 'knorr_mean', 'dena_extreme1',
+                            'dena_extreme2', 'knorr_extreme1',
+                            'knorr_extreme2', 'knorr_extreme3']
     if curve_name.split('_')[0] not in ['dena', 'knorr']:
         raise ValueError("Wrong wind efficiency curve name. Must be one of " +
-                         "the following: 'dena_mean', 'knorr_mean' " +  # TODO add names
+                         "the following: {}".format(possible_curve_names) +
                          "but is {}".format(curve_name))
     path = os.path.join(os.path.dirname(__file__), 'data',
                         'wind_efficiency_curves_{}.csv'.format(
@@ -99,8 +102,8 @@ def get_wind_efficiency_curve(curve_name='dena_mean'):
     # Raise error if wind efficiency curve specified in 'curve_name' does not
     # exist
     if curve_name not in list(wind_efficiency_curves):
-        raise ValueError("Efficiency curve name does not exist. Must be " +
-                         "'dena_mean', 'knorr_mean' " +  # TODO add names
+        raise ValueError("Efficiency curve name does not exist. Must be one" +
+                         "of the following: {}".format(possible_curve_names) +
                          "but is {}".format(curve_name))
     efficiency_curve = wind_efficiency_curves[['wind_speed', curve_name]]
     efficiency_curve.columns = ['wind_speed', 'efficiency']