use separate classes for version checking

pvlib · wholmgren · Sep 10, 2020 · Aug 13, 2018 · Aug 14, 2018 · Mar 22, 2020
commit 008f8ca8bb92a5fe6ccdb7cb37027083e2d602d7
diff --git a/benchmarks/irradiance.py b/benchmarks/irradiance.py
@@ -6,7 +6,7 @@
 from pvlib import irradiance, location
 
 
-class TimeSuite:
+class Irradiance:
 
     def setup(self):
         self.times = pd.date_range(start='20180601', freq='1min',

diff --git a/benchmarks/location.py b/benchmarks/location.py
@@ -7,16 +7,20 @@
 from pkg_resources import parse_version
 
 
-class TimeSuite:
+def set_solar_position(obj):
+    obj.location = pvlib.location.Location(32, -110, altitude=700,
+                                           tz='Etc/GMT+7')
+    obj.times = pd.date_range(start='20180601', freq='3min',
+                              periods=1440)
+    obj.days = pd.date_range(start='20180101', freq='d', periods=365,
+                             tz=obj.location.tz)
+    obj.solar_position = obj.location.get_solarposition(obj.times)
+
+
+class Location:
 
     def setup(self):
-        self.location = pvlib.location.Location(32, -110, altitude=700,
-                                                tz='Etc/GMT+7')
-        self.times = pd.date_range(start='20180601', freq='3min',
-                                   periods=1440)
-        self.days = pd.date_range(start='20180101', freq='d', periods=365,
-                                  tz=self.location.tz)
-        self.solar_position = self.location.get_solarposition(self.times)
+        set_solar_position(self)
 
     # GH 502
     def time_location_get_airmass(self):
@@ -29,12 +33,19 @@ def time_location_get_clearsky(self):
         self.location.get_clearsky(times=self.times,
                                    solar_position=self.solar_position)
 
-    if parse_version(pvlib.__version__) >= parse_version('0.6.1'):
-        def time_location_get_sun_rise_set_transit_pyephem(self):
-            self.location.get_sun_rise_set_transit(times=self.days,
-                                                   method='pyephem')
 
-    if parse_version(pvlib.__version__) >= parse_version('0.6.1'):
-        def time_location_get_sun_rise_set_transit_spa(self):
-            self.location.get_sun_rise_set_transit(times=self.days,
-                                                   method='spa')
+class Location_0_6_1:
+
+    def setup(self):
+        if parse_version(pvlib.__version__) < parse_version('0.6.1'):
+            raise NotImplementedError
+
+        set_solar_position(self)
+
+    def time_location_get_sun_rise_set_transit_pyephem(self):
+        self.location.get_sun_rise_set_transit(times=self.days,
+                                               method='pyephem')
+
+    def time_location_get_sun_rise_set_transit_spa(self):
+        self.location.get_sun_rise_set_transit(times=self.days,
+                                               method='spa')
diff --git a/benchmarks/solarposition.py b/benchmarks/solarposition.py
@@ -15,7 +15,7 @@
     sun_rise_set_transit_spa = solarposition.get_sun_rise_set_transit
 
 
-class TimeSuite:
+class SolarPosition:
 
     def setup(self):
         self.times = pd.date_range(start='20180601', freq='1min',

diff --git a/benchmarks/temperature.py b/benchmarks/temperature.py
@@ -0,0 +1,50 @@
+"""
+ASV benchmarks for irradiance.py
+"""
+
+import pandas as pd
+import pvlib
+from pkg_resources import parse_version
+import functools
+
+
+def set_weather_data(obj):
+    obj.times = pd.date_range(start='20180601', freq='1min',
+                              periods=14400)
+    obj.poa = pd.Series(1000, index=obj.times)
+    obj.tamb = pd.Series(20, index=obj.times)
+    obj.windspeed = pd.Series(2, index=obj.times)
+
+
+if parse_version(pvlib.__version__) >= parse_version('0.7.0'):
+    params = pvlib.temperature.TEMPERATURE_MODEL_PARAMETERS['sapm']
+    params = params['open_rack_glass_glass']
+    sapm_cell = functools.partial(pvlib.temperature.sapm_cell, **params)
+else:
+    sapm_celltemp = pvlib.pvsystem.sapm_celltemp
+
+    def sapm_cell(poa_global, temp_air, wind_speed):
+        return sapm_celltemp(poa_global, wind_speed, temp_air)
+
+
+class SAPM:
+
+    def setup(self):
+        set_weather_data(self)
+
+    def time_sapm_cell(self):
+        # use version-appropriate wrapper
+        sapm_cell(self.poa, self.tamb, self.windspeed)
+
+
+class Fuentes:

+    def setup(self):
+        if parse_version(pvlib.__version__) < parse_version('0.8.0'):
+            raise NotImplementedError
+
+        set_weather_data(self)
+
+    def time_fuentes(self):
+        pvlib.temperature.fuentes(self.poa, self.tamb, self.wind_speed,
+                                  noct_installed=45)
diff --git a/benchmarks/tracking.py b/benchmarks/tracking.py
@@ -7,7 +7,7 @@
 import numpy as np
 
 
-class TimeSuite:
+class SingleAxis:
 
     def setup(self):
         self.times = pd.date_range(start='20180601', freq='1min',