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
| import pytest
|
| from pandas import (
| Interval,
| Period,
| Timestamp,
| )
|
|
| class TestIntervalConstructors:
| @pytest.mark.parametrize(
| "left, right",
| [
| ("a", "z"),
| (("a", "b"), ("c", "d")),
| (list("AB"), list("ab")),
| (Interval(0, 1), Interval(1, 2)),
| (Period("2018Q1", freq="Q"), Period("2018Q1", freq="Q")),
| ],
| )
| def test_construct_errors(self, left, right):
| # GH#23013
| msg = "Only numeric, Timestamp and Timedelta endpoints are allowed"
| with pytest.raises(ValueError, match=msg):
| Interval(left, right)
|
| def test_constructor_errors(self):
| msg = "invalid option for 'closed': foo"
| with pytest.raises(ValueError, match=msg):
| Interval(0, 1, closed="foo")
|
| msg = "left side of interval must be <= right side"
| with pytest.raises(ValueError, match=msg):
| Interval(1, 0)
|
| @pytest.mark.parametrize(
| "tz_left, tz_right", [(None, "UTC"), ("UTC", None), ("UTC", "US/Eastern")]
| )
| def test_constructor_errors_tz(self, tz_left, tz_right):
| # GH#18538
| left = Timestamp("2017-01-01", tz=tz_left)
| right = Timestamp("2017-01-02", tz=tz_right)
|
| if tz_left is None or tz_right is None:
| error = TypeError
| msg = "Cannot compare tz-naive and tz-aware timestamps"
| else:
| error = ValueError
| msg = "left and right must have the same time zone"
| with pytest.raises(error, match=msg):
| Interval(left, right)
|
|
