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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
import time
from datetime import datetime
import pytest
from arrow import util
class TestUtil:
def test_next_weekday(self):
# Get first Monday after epoch
assert util.next_weekday(datetime(1970, 1, 1), 0) == datetime(1970, 1, 5)
# Get first Tuesday after epoch
assert util.next_weekday(datetime(1970, 1, 1), 1) == datetime(1970, 1, 6)
# Get first Wednesday after epoch
assert util.next_weekday(datetime(1970, 1, 1), 2) == datetime(1970, 1, 7)
# Get first Thursday after epoch
assert util.next_weekday(datetime(1970, 1, 1), 3) == datetime(1970, 1, 1)
# Get first Friday after epoch
assert util.next_weekday(datetime(1970, 1, 1), 4) == datetime(1970, 1, 2)
# Get first Saturday after epoch
assert util.next_weekday(datetime(1970, 1, 1), 5) == datetime(1970, 1, 3)
# Get first Sunday after epoch
assert util.next_weekday(datetime(1970, 1, 1), 6) == datetime(1970, 1, 4)
# Weekdays are 0-indexed
with pytest.raises(ValueError):
util.next_weekday(datetime(1970, 1, 1), 7)
with pytest.raises(ValueError):
util.next_weekday(datetime(1970, 1, 1), -1)
def test_is_timestamp(self):
timestamp_float = time.time()
timestamp_int = int(timestamp_float)
assert util.is_timestamp(timestamp_int)
assert util.is_timestamp(timestamp_float)
assert util.is_timestamp(str(timestamp_int))
assert util.is_timestamp(str(timestamp_float))
assert not util.is_timestamp(True)
assert not util.is_timestamp(False)
class InvalidTimestamp:
pass
assert not util.is_timestamp(InvalidTimestamp())
full_datetime = "2019-06-23T13:12:42"
assert not util.is_timestamp(full_datetime)
def test_validate_ordinal(self):
timestamp_float = 1607066816.815537
timestamp_int = int(timestamp_float)
timestamp_str = str(timestamp_int)
with pytest.raises(TypeError):
util.validate_ordinal(timestamp_float)
with pytest.raises(TypeError):
util.validate_ordinal(timestamp_str)
with pytest.raises(TypeError):
util.validate_ordinal(True)
with pytest.raises(TypeError):
util.validate_ordinal(False)
with pytest.raises(ValueError):
util.validate_ordinal(timestamp_int)
with pytest.raises(ValueError):
util.validate_ordinal(-1 * timestamp_int)
with pytest.raises(ValueError):
util.validate_ordinal(0)
try:
util.validate_ordinal(1)
except (ValueError, TypeError) as exp:
pytest.fail(f"Exception raised when shouldn't have ({type(exp)}).")
try:
util.validate_ordinal(datetime.max.toordinal())
except (ValueError, TypeError) as exp:
pytest.fail(f"Exception raised when shouldn't have ({type(exp)}).")
ordinal = datetime.utcnow().toordinal()
ordinal_str = str(ordinal)
ordinal_float = float(ordinal) + 0.5
with pytest.raises(TypeError):
util.validate_ordinal(ordinal_str)
with pytest.raises(TypeError):
util.validate_ordinal(ordinal_float)
with pytest.raises(ValueError):
util.validate_ordinal(-1 * ordinal)
try:
util.validate_ordinal(ordinal)
except (ValueError, TypeError) as exp:
pytest.fail(f"Exception raised when shouldn't have ({type(exp)}).")
full_datetime = "2019-06-23T13:12:42"
class InvalidOrdinal:
pass
with pytest.raises(TypeError):
util.validate_ordinal(InvalidOrdinal())
with pytest.raises(TypeError):
util.validate_ordinal(full_datetime)
def test_normalize_timestamp(self):
timestamp = 1591161115.194556
millisecond_timestamp = 1591161115194
microsecond_timestamp = 1591161115194556
assert util.normalize_timestamp(timestamp) == timestamp
assert util.normalize_timestamp(millisecond_timestamp) == 1591161115.194
assert util.normalize_timestamp(microsecond_timestamp) == 1591161115.194556
with pytest.raises(ValueError):
util.normalize_timestamp(3e17)
def test_iso_gregorian(self):
with pytest.raises(ValueError):
util.iso_to_gregorian(2013, 0, 5)
with pytest.raises(ValueError):
util.iso_to_gregorian(2013, 8, 0)
|
