9
9
import pandas .lib as lib
10
10
import pandas ._period as period
11
11
import pandas .algos as algos
12
- from pandas .tseries .holiday import Holiday , SA , next_monday ,USMartinLutherKingJr ,USMemorialDay ,AbstractHolidayCalendar
12
+ from pandas .tseries .holiday import Holiday , SA , next_monday ,USMartinLutherKingJr ,USMemorialDay ,AbstractHolidayCalendar
13
13
import datetime
14
-
15
-
16
14
from pandas import DateOffset
17
15
18
16
@@ -378,9 +376,9 @@ def test_get_reverse_indexer():
378
376
379
377
380
378
def test_pad_backfill_object_segfault ():
381
- from datetime import datetime
379
+
382
380
old = np .array ([], dtype = 'O' )
383
- new = np .array ([datetime (2010 , 12 , 31 )], dtype = 'O' )
381
+ new = np .array ([datetime . datetime (2010 , 12 , 31 )], dtype = 'O' )
384
382
385
383
result = algos .pad_object (old , new )
386
384
expected = np .array ([- 1 ], dtype = np .int64 )
@@ -644,13 +642,13 @@ def test_unicode(self):
644
642
pass
645
643
646
644
def test_datetime (self ):
647
- import datetime
645
+
648
646
dates = [datetime .datetime (2012 , 1 , x ) for x in range (1 , 20 )]
649
647
index = Index (dates )
650
648
self .assertEqual (index .inferred_type , 'datetime64' )
651
649
652
650
def test_date (self ):
653
- import datetime
651
+
654
652
dates = [datetime .date (2012 , 1 , x ) for x in range (1 , 20 )]
655
653
index = Index (dates )
656
654
self .assertEqual (index .inferred_type , 'date' )
@@ -755,6 +753,7 @@ def test_get_period_field_array_raises_on_out_of_range(self):
755
753
self .assertRaises (ValueError , period .get_period_field_arr , - 1 , np .empty (1 ), 0 )
756
754
757
755
class TestFederalHolidayCalendar (tm .TestCase ):
756
+
758
757
# Test for issue 10278
759
758
def test_no_mlk_before_1984 (self ):
760
759
class MLKCalendar (AbstractHolidayCalendar ):
@@ -768,11 +767,11 @@ class MemorialDay(AbstractHolidayCalendar):
768
767
rules = [USMemorialDay ]
769
768
holidays = MemorialDay ().holidays (start = '1971' , end = '1980' ).to_pydatetime ().tolist ()
770
769
# Fixes 5/31 error and checked manually against wikipedia
771
- self .assertEqual (holidays , [datetime .datetime (1971 , 5 , 31 , 0 , 0 ), datetime .datetime (1972 , 5 , 29 , 0 , 0 ),
772
- datetime .datetime (1973 , 5 , 28 , 0 , 0 ), datetime .datetime (1974 , 5 , 27 , 0 , 0 ),
773
- datetime .datetime (1975 , 5 , 26 , 0 , 0 ), datetime .datetime (1976 , 5 , 31 , 0 , 0 ),
774
- datetime .datetime (1977 , 5 , 30 , 0 , 0 ), datetime .datetime (1978 , 5 , 29 , 0 , 0 ),
775
- datetime .datetime (1979 , 5 , 28 , 0 , 0 )])
770
+ self .assertEqual (holidays , [datetime .datetime (1971 , 5 , 31 , 0 , 0 ), datetime .datetime (1972 , 5 , 29 , 0 , 0 ),
771
+ datetime .datetime (1973 , 5 , 28 , 0 , 0 ), datetime .datetime (1974 , 5 , 27 , 0 , 0 ),
772
+ datetime .datetime (1975 , 5 , 26 , 0 , 0 ), datetime .datetime (1976 , 5 , 31 , 0 , 0 ),
773
+ datetime .datetime (1977 , 5 , 30 , 0 , 0 ), datetime .datetime (1978 , 5 , 29 , 0 , 0 ),
774
+ datetime .datetime (1979 , 5 , 28 , 0 , 0 )])
776
775
777
776
778
777
0 commit comments