2
2
3
3
import numpy as np
4
4
5
- import pandas as pd
6
5
import pandas .util .testing as tm
7
6
from pandas import (TimedeltaIndex , timedelta_range , Int64Index , Float64Index ,
8
- Index , Timedelta )
7
+ Index , Timedelta , NaT )
9
8
10
- from ..datetimelike import DatetimeLike
11
9
12
-
13
- class TestTimedeltaIndex (DatetimeLike ):
14
- _holder = TimedeltaIndex
10
+ class TestTimedeltaIndex (object ):
15
11
_multiprocess_can_split_ = True
16
12
17
- def test_numeric_compat (self ):
18
- # Dummy method to override super's version; this test is now done
19
- # in test_arithmetic.py
20
- pass
21
-
22
- def setup_method (self , method ):
23
- self .indices = dict (index = tm .makeTimedeltaIndex (10 ))
24
- self .setup_indices ()
25
-
26
- def create_index (self ):
27
- return pd .to_timedelta (range (5 ), unit = 'd' ) + pd .offsets .Hour (1 )
28
-
29
13
def test_astype (self ):
30
14
# GH 13149, GH 13209
31
- idx = TimedeltaIndex ([1e14 , 'NaT' , pd . NaT , np .NaN ])
15
+ idx = TimedeltaIndex ([1e14 , 'NaT' , NaT , np .NaN ])
32
16
33
17
result = idx .astype (object )
34
- expected = Index ([Timedelta ('1 days 03:46:40' )] + [pd . NaT ] * 3 ,
18
+ expected = Index ([Timedelta ('1 days 03:46:40' )] + [NaT ] * 3 ,
35
19
dtype = object )
36
20
tm .assert_index_equal (result , expected )
37
21
@@ -51,7 +35,7 @@ def test_astype(self):
51
35
52
36
def test_astype_timedelta64 (self ):
53
37
# GH 13149, GH 13209
54
- idx = TimedeltaIndex ([1e14 , 'NaT' , pd . NaT , np .NaN ])
38
+ idx = TimedeltaIndex ([1e14 , 'NaT' , NaT , np .NaN ])
55
39
56
40
result = idx .astype ('timedelta64' )
57
41
expected = Float64Index ([1e+14 ] + [np .NaN ] * 3 , dtype = 'float64' )
@@ -69,28 +53,7 @@ def test_astype_timedelta64(self):
69
53
float , 'datetime64' , 'datetime64[ns]' ])
70
54
def test_astype_raises (self , dtype ):
71
55
# GH 13149, GH 13209
72
- idx = TimedeltaIndex ([1e14 , 'NaT' , pd . NaT , np .NaN ])
56
+ idx = TimedeltaIndex ([1e14 , 'NaT' , NaT , np .NaN ])
73
57
msg = 'Cannot cast TimedeltaIndex to dtype'
74
58
with tm .assert_raises_regex (TypeError , msg ):
75
59
idx .astype (dtype )
76
-
77
- def test_pickle_compat_construction (self ):
78
- pass
79
-
80
- def test_shift (self ):
81
- # test shift for TimedeltaIndex
82
- # err8083
83
-
84
- drange = self .create_index ()
85
- result = drange .shift (1 )
86
- expected = TimedeltaIndex (['1 days 01:00:00' , '2 days 01:00:00' ,
87
- '3 days 01:00:00' ,
88
- '4 days 01:00:00' , '5 days 01:00:00' ],
89
- freq = 'D' )
90
- tm .assert_index_equal (result , expected )
91
-
92
- result = drange .shift (3 , freq = '2D 1s' )
93
- expected = TimedeltaIndex (['6 days 01:00:03' , '7 days 01:00:03' ,
94
- '8 days 01:00:03' , '9 days 01:00:03' ,
95
- '10 days 01:00:03' ], freq = 'D' )
96
- tm .assert_index_equal (result , expected )
0 commit comments