add SingleAxisTracker support to ModelChain

Author: wholmgren

pvlib/modelchain.py

@@ -6,9 +6,12 @@
 the time to read the source code for the module.
 """
 
+from functools import partial
+
 import pandas as pd
 
 from pvlib import solarposition, pvsystem, clearsky, atmosphere
+from pvlib.tracking import SingleAxisTracker
 import pvlib.irradiance  # avoid name conflict with full import
 
 
@@ -317,9 +320,32 @@ def run_model(self, times, irradiance=None, weather=None):
                 airmass_data=self.airmass['airmass_absolute'])
         self.irradiance = irradiance
 
-        self.total_irrad = self.system.get_irradiance(
-            self.solar_position['apparent_zenith'],
-            self.solar_position['azimuth'],
+        # PVSystem.get_irradiance and SingleAxisTracker.get_irradiance
+        # have different method signatures, so use partial to handle
+        # the differences.
+        if isinstance(self.system, SingleAxisTracker):
+            self.tracking = self.system.singleaxis(
+                self.solar_position['apparent_zenith'],
+                self.solar_position['azimuth'])
+            self.tracking['surface_tilt'] = (
+                self.tracking['surface_tilt']
+                    .fillna(self.system.axis_tilt))
+            self.tracking['surface_azimuth'] = (
+                self.tracking['surface_azimuth']
+                    .fillna(self.system.axis_azimuth))
+            get_irradiance = partial(
+                self.system.get_irradiance,
+                surface_tilt=self.tracking['surface_tilt'],
+                surface_azimuth=self.tracking['surface_azimuth'],
+                solar_zenith=self.solar_position['apparent_zenith'],
+                solar_azimuth=self.solar_position['azimuth'])
+        else:
+            get_irradiance = partial(
+                self.system.get_irradiance,
+                self.solar_position['apparent_zenith'],
+                self.solar_position['azimuth'])
+                
+        self.total_irrad = get_irradiance(
             self.irradiance['dni'],
             self.irradiance['ghi'],
             self.irradiance['dhi'],

pvlib/test/test_modelchain.py

@@ -4,6 +4,7 @@
 from pvlib import modelchain, pvsystem
 from pvlib.modelchain import ModelChain
 from pvlib.pvsystem import PVSystem
+from pvlib.tracking import SingleAxisTracker
 from pvlib.location import Location
 
 from pandas.util.testing import assert_series_equal, assert_frame_equal
@@ -100,6 +101,19 @@ def test_run_model_with_weather():
     assert_series_equal(ac, expected)
 
 
+def test_run_model_tracker():
+    system, location = mc_setup()
+    system = SingleAxisTracker(module_parameters=system.module_parameters,
+                               inverter_parameters=system.inverter_parameters)
+    mc = ModelChain(system, location)
+    times = pd.date_range('20160101 1200-0700', periods=2, freq='6H')
+    ac = mc.run_model(times).ac
+
+    expected = pd.Series(np.array([  121.421719,  -2.00000000e-02]),
+                         index=times)
+    assert_series_equal(ac, expected)
+    
+
 @raises(ValueError)
 def test_bad_get_orientation():
     modelchain.get_orientation('bad value')