1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
| import numpy as np
| import pytest
|
| from pandas import (
| Index,
| Series,
| Timedelta,
| timedelta_range,
| )
| import pandas._testing as tm
|
|
| class TestTimedeltaIndex:
| def test_misc_coverage(self):
| rng = timedelta_range("1 day", periods=5)
| result = rng.groupby(rng.days)
| assert isinstance(next(iter(result.values()))[0], Timedelta)
|
| def test_map(self):
| # test_map_dictlike generally tests
|
| rng = timedelta_range("1 day", periods=10)
|
| f = lambda x: x.days
| result = rng.map(f)
| exp = Index([f(x) for x in rng], dtype=np.int64)
| tm.assert_index_equal(result, exp)
|
| def test_fields(self):
| rng = timedelta_range("1 days, 10:11:12.100123456", periods=2, freq="s")
| tm.assert_index_equal(rng.days, Index([1, 1], dtype=np.int64))
| tm.assert_index_equal(
| rng.seconds,
| Index([10 * 3600 + 11 * 60 + 12, 10 * 3600 + 11 * 60 + 13], dtype=np.int32),
| )
| tm.assert_index_equal(
| rng.microseconds,
| Index([100 * 1000 + 123, 100 * 1000 + 123], dtype=np.int32),
| )
| tm.assert_index_equal(rng.nanoseconds, Index([456, 456], dtype=np.int32))
|
| msg = "'TimedeltaIndex' object has no attribute '{}'"
| with pytest.raises(AttributeError, match=msg.format("hours")):
| rng.hours
| with pytest.raises(AttributeError, match=msg.format("minutes")):
| rng.minutes
| with pytest.raises(AttributeError, match=msg.format("milliseconds")):
| rng.milliseconds
|
| # with nat
| s = Series(rng)
| s[1] = np.nan
|
| tm.assert_series_equal(s.dt.days, Series([1, np.nan], index=[0, 1]))
| tm.assert_series_equal(
| s.dt.seconds, Series([10 * 3600 + 11 * 60 + 12, np.nan], index=[0, 1])
| )
|
| # preserve name (GH15589)
| rng.name = "name"
| assert rng.days.name == "name"
|
|
