add tests for __repr__() functions

Author: JohannesOos

pvlib/test/test_location.py

@@ -160,3 +160,10 @@ def test_get_airmass_valueerror():
                              end='20160101T1800-0700',
                              freq='3H')
     clearsky = tus.get_airmass(times, model='invalid_model')
+    
+def test_Location___repr__():
+    tus = Location(32.2, -111, 'US/Arizona', 700, 'Tucson')
+    assert tus.__repr__()==('Tucson: latitude=32.2, longitude=-111, '+
+    'tz=US/Arizona, altitude=700')
+    
+    

pvlib/test/test_modelchain.py

@@ -198,3 +198,18 @@ def test_basic_chain_altitude_pressure():
     expected = pd.Series(np.array([  1.15771428788e+02,  -2.00000000e-02]),
                          index=times)
     assert_series_equal(ac, expected)
+    
+    
+def test_ModelChain___repr__():
+    system = PVSystem()
+    location = Location(32.2, -111, altitude=700)
+    strategy = 'south_at_latitude_tilt'
+
+    mc = ModelChain(system, location, orientation_strategy=strategy)
+
+    # the || accounts for the coercion of 'None' to None
+    assert mc.__repr__() == ('ModelChain for: PVSystem with tilt:32.2 and '+
+    'azimuth: 180 with Module: None and Inverter: None with '+
+    'orientation_startegy: south_at_latitude_tilt clearsky_model:ineichen '+
+    'transposition_model: haydavies solar_position_method: nrel_numpy '+
+    'airmass_model: kastenyoung1989')

pvlib/test/test_pvsystem.py

@@ -455,6 +455,21 @@ def test_PVSystem_localize_with_latlon():
     assert localized_system.longitude == -111
 
 
+def test_PVSystem___repr__():
+    system = pvsystem.PVSystem(module='blah', inverter='blarg')
+
+    assert system.__repr__()==('PVSystem with tilt:0 and azimuth:'+
+    ' 180 with Module: blah and Inverter: blarg')
+    
+
+def test_PVSystem_localize___repr__():
+    system = pvsystem.PVSystem(module='blah', inverter='blarg')
+    localized_system = system.localize(latitude=32, longitude=-111)
+
+    assert localized_system.__repr__()==('LocalizedPVSystem with tilt:0 and'+
+    ' azimuth: 180 with Module: blah and Inverter: blarg and Location: None:'+
+    ' latitude=32, longitude=-111, tz=UTC, altitude=0')
+
 # we could retest each of the models tested above
 # when they are attached to LocalizedPVSystem, but
 # that's probably not necessary at this point.
@@ -470,3 +485,13 @@ def test_LocalizedPVSystem_creation():
     assert localized_system.inverter == 'blarg'
     assert localized_system.latitude == 32
     assert localized_system.longitude == -111
+
+    
+def test_LocalizedPVSystem___repr__():
+    localized_system = pvsystem.LocalizedPVSystem(latitude=32,
+                                                  longitude=-111,
+                                                  module='blah',
+                                                  inverter='blarg')
+
+    assert localized_system.__repr__()==('LocalizedPVSystem with tilt:0 and'+
+    ' azimuth: 180 with Module: blah and Inverter: blarg and Location: None')