@@ -224,8 +224,9 @@ def test_lookup_linke_turbidity_nointerp():
224224
225225
226226def  test_lookup_linke_turbidity_months ():
227-     times  =  pd .date_range (start = '2014-04-01' , end = '2014-07-01' ,
228-                           freq = '1M' , tz = 'America/Phoenix' )
227+     times  =  pd .date_range (start = '2014-05-01' , end = '2014-07-01' ,
228+                           freq = '1MS' , tz = 'America/Phoenix' ,
229+                           ) -  pd .Timedelta (days = 1 )
229230    expected  =  pd .Series (
230231        np .array ([2.89918032787 , 2.97540983607 , 3.19672131148 ]), index = times 
231232    )
@@ -234,8 +235,9 @@ def test_lookup_linke_turbidity_months():
234235
235236
236237def  test_lookup_linke_turbidity_months_leapyear ():
237-     times  =  pd .date_range (start = '2016-04-01' , end = '2016-07-01' ,
238-                           freq = '1M' , tz = 'America/Phoenix' )
238+     times  =  pd .date_range (start = '2016-05-01' , end = '2016-07-01' ,
239+                           freq = '1MS' , tz = 'America/Phoenix' ,
240+                           ) -  pd .Timedelta (days = 1 )
239241    expected  =  pd .Series (
240242        np .array ([2.89918032787 , 2.97540983607 , 3.19672131148 ]), index = times 
241243    )
@@ -245,14 +247,16 @@ def test_lookup_linke_turbidity_months_leapyear():
245247
246248def  test_lookup_linke_turbidity_nointerp_months ():
247249    times  =  pd .date_range (start = '2014-04-10' , end = '2014-07-10' ,
248-                           freq = '1M' , tz = 'America/Phoenix' )
250+                           freq = '1MS' , tz = 'America/Phoenix' ,
251+                           ) -  pd .Timedelta (days = 1 )
249252    expected  =  pd .Series (np .array ([2.85 , 2.95 , 3. ]), index = times )
250253    out  =  clearsky .lookup_linke_turbidity (times , 32.125 , - 110.875 ,
251254                                          interp_turbidity = False )
252255    assert_series_equal (expected , out )
253256    # changing the dates shouldn't matter if interp=False 
254257    times  =  pd .date_range (start = '2014-04-05' , end = '2014-07-05' ,
255-                           freq = '1M' , tz = 'America/Phoenix' )
258+                           freq = '1MS' , tz = 'America/Phoenix' ,
259+                           ) -  pd .Timedelta (days = 1 )
256260    out  =  clearsky .lookup_linke_turbidity (times , 32.125 , - 110.875 ,
257261                                          interp_turbidity = False )
258262    assert_series_equal (expected , out )
0 commit comments