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
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
|
import cf
import datetime
import numpy
import os
import time
import unittest
class TimeDurationTest(unittest.TestCase):
filename = os.path.join(os.path.dirname(os.path.abspath(__file__)),
'test_file.nc')
chunk_sizes = (17, 34, 300, 100000)[::-1]
def test_TimeDuration(self):
self.assertTrue(cf.TimeDuration(2, 'calendar_years') > cf.TimeDuration(1, 'calendar_years'))
self.assertTrue(cf.TimeDuration(2, 'calendar_years') < cf.TimeDuration(25, 'calendar_months'))
self.assertTrue(cf.TimeDuration(2, 'hours') <= cf.TimeDuration(1, 'days'))
self.assertTrue(cf.TimeDuration(2, 'hours') == cf.TimeDuration(1/12.0, 'days'))
self.assertTrue(cf.TimeDuration(2, 'days') == cf.TimeDuration(48, 'hours'))
self.assertTrue(cf.TimeDuration(2, 'days') == cf.Data(2))
# self.assertTrue(cf.TimeDuration(2, 'days') > cf.Data(1.5, ''))
# self.assertTrue(cf.TimeDuration(2, 'days') > cf.Data(1.5, '1'))
# self.assertTrue(cf.TimeDuration(2, 'days') < cf.Data(0.03, '100'))
self.assertTrue(cf.TimeDuration(2, 'days') == cf.Data([2.], 'days'))
self.assertTrue(cf.TimeDuration(2, 'days') > cf.Data([[60]], 'seconds'))
self.assertTrue(cf.TimeDuration(2, 'hours') <= 2)
self.assertTrue(cf.TimeDuration(2, 'days') != 30.5)
self.assertTrue(cf.TimeDuration(2, 'calendar_years') > numpy.array(1.5))
self.assertTrue(cf.TimeDuration(2, 'calendar_months') < numpy.array([[12]]))
self.assertFalse(cf.TimeDuration(2, 'calendar_years') <= cf.TimeDuration(1, 'calendar_years'))
self.assertFalse(cf.TimeDuration(2, 'calendar_years') >= cf.TimeDuration(25, 'calendar_months'))
self.assertFalse(cf.TimeDuration(2, 'hours') > cf.TimeDuration(1, 'days'))
self.assertFalse(cf.TimeDuration(2, 'hours') != cf.TimeDuration(1/12.0, 'days'))
self.assertFalse(cf.TimeDuration(2, 'days') != cf.TimeDuration(48, 'hours'))
self.assertFalse(cf.TimeDuration(2, 'days') != cf.Data(2))
# self.assertFalse(cf.TimeDuration(2, 'days') <= cf.Data(1.5, ''))
# self.assertFalse(cf.TimeDuration(2, 'days') <= cf.Data(1.5, '1'))
# self.assertFalse(cf.TimeDuration(2, 'days') >= cf.Data(0.03, '100'))
self.assertFalse(cf.TimeDuration(2, 'days') != cf.Data([2.], 'days'))
self.assertFalse(cf.TimeDuration(2, 'days') <= cf.Data([[60]], 'seconds'))
self.assertFalse(cf.TimeDuration(2, 'hours') > 2)
self.assertFalse(cf.TimeDuration(2, 'days') == 30.5)
self.assertFalse(cf.TimeDuration(2, 'calendar_years') <= numpy.array(1.5))
self.assertFalse(cf.TimeDuration(2, 'calendar_months') >= numpy.array([[12]]))
self.assertTrue(cf.TimeDuration(64, 'calendar_years') + 2 == cf.Y(66))
self.assertTrue(cf.TimeDuration(64, 'calendar_years') - 2.5 == cf.Y(61.5))
self.assertTrue(cf.M(23) + cf.TimeDuration(64, 'calendar_years') == cf.M(791))
self.assertTrue(cf.TimeDuration(64, 'calendar_years') + cf.M(24) == cf.Y(66))
self.assertTrue(cf.TimeDuration(36, 'calendar_months') / numpy.array(8) == cf.M(4))
self.assertTrue(cf.TimeDuration(36, 'calendar_months') / numpy.array(8.0) == cf.M(36/8.0))
self.assertTrue(cf.TimeDuration(12, 'calendar_months') * cf.Data([[1.5]]) == cf.Y(1.5))
self.assertTrue(cf.TimeDuration(36, 'calendar_months') // cf.Data([0.825], '10') == cf.M(4.3))
self.assertTrue(cf.TimeDuration(36, 'calendar_months') % 10 == cf.M(6))
self.assertTrue(cf.TimeDuration(24, 'hours') + cf.TimeDuration(0.5, 'days') == cf.h(36.0))
self.assertTrue(cf.TimeDuration(0.5, 'days') + cf.TimeDuration(24, 'hours') == cf.D(1.5))
t = cf.TimeDuration(24, 'hours')
t += 2
self.assertTrue(t == cf.h(26))
t -= cf.Data(3, 'hours')
self.assertTrue(t == cf.h(23))
t = cf.TimeDuration(24.0, 'hours')
t += 2
self.assertTrue(t == cf.h(26))
t -= cf.Data(2.5, 'hours')
self.assertTrue(t == cf.h(23.5))
t *= 2
self.assertTrue(t == cf.h(47.0))
t -= 0.5
self.assertTrue(t == cf.h(46.5))
t /= 3
self.assertTrue(t == cf.h(15.5))
t += 5.5
self.assertTrue(t == cf.h(21.0))
t //= numpy.array(2)
self.assertTrue(t == cf.h(10.0))
t *= 10
self.assertTrue(t == cf.h(100.0))
t %= 3
self.assertTrue(t == cf.h(1.0))
self.assertTrue(cf.M().interval(1999, 12) ==
(cf.dt('1999-12-01 00:00:00'), cf.dt('2000-01-01 00:00:00')))
self.assertTrue(cf.Y(2).interval(2000, 2, end=True) ==
(cf.dt('1998-02-01 00:00:00'), cf.dt('2000-02-01 00:00:00')))
self.assertTrue(cf.D(30).interval(1983, 12, 1, 6) ==
(cf.dt('1983-12-01 06:00:00'), cf.dt('1983-12-31 06:00:00')))
self.assertTrue(cf.D(30).interval(1983, 12, 1, 6, end=True) ==
(cf.dt('1983-11-01 06:00:00'), cf.dt('1983-12-01 06:00:00')))
self.assertTrue(cf.D(0).interval(1984, 2, 3) ==
(cf.dt('1984-02-03 00:00:00'), cf.dt('1984-02-03 00:00:00')))
self.assertTrue(cf.D(5, hour=6).interval(2004, 3, 2, end=True) ==
(cf.dt('2004-02-26 06:00:00'), cf.dt('2004-03-02 06:00:00')))
self.assertTrue(cf.D(5, hour=6).interval(2004, 3, 2, end=True, calendar='noleap') ==
(cf.dt('2004-02-25 06:00:00'), cf.dt('2004-03-02 06:00:00')))
self.assertTrue(cf.D(5, hour=6).interval(2004, 3, 2, end=True, calendar='360_day') ==
(cf.dt('2004-02-27 06:00:00'), cf.dt('2004-03-02 06:00:00')))
self.assertTrue(cf.h(19897.5).interval(1984, 2, 3, 0) ==
(cf.dt('1984-02-03 00:00:00'), cf.dt('1986-05-12 01:30:00')))
self.assertTrue(cf.h(19897.546).interval(1984, 2, 3, 0, end=True) ==
(cf.dt('1981-10-26 22:27:14'), cf.dt('1984-02-03 00:00:00')))
#--- End: def
def test_TimeDuration_bounds(self):
for direction in (True, False):
for x, y in zip(
[cf.Y().bounds(1984, 1, 1, direction=direction),
cf.Y().bounds(1984, 12, 1, direction=direction),
cf.Y().bounds(1984, 12, 3, direction=direction),
cf.Y(month=9).bounds(1984, 1, 1, direction=direction),
cf.Y(month=9).bounds(1984, 3, 3, direction=direction),
cf.Y(month=9).bounds(1984, 9, 20, direction=direction),
cf.Y(month=9, day=13).bounds(1984, 12, 12, direction=direction),
],
[(cf.dt('1984-01-01'), cf.dt('1985-01-01')),
(cf.dt('1984-01-01'), cf.dt('1985-01-01')),
(cf.dt('1984-01-01'), cf.dt('1985-01-01')),
(cf.dt('1983-09-01'), cf.dt('1984-09-01')),
(cf.dt('1983-09-01'), cf.dt('1984-09-01')),
(cf.dt('1984-09-01'), cf.dt('1985-09-01')),
(cf.dt('1984-09-13'), cf.dt('1985-09-13')),
]):
if direction is False:
y = y[::-1]
self.assertTrue(x==y, "{}!={}".format(x, y))
for x, y in zip(
[cf.M().bounds(1984, 1, 1, direction=direction),
cf.M().bounds(1984, 12, 1, direction=direction),
cf.M().bounds(1984, 12, 3, direction=direction),
cf.M(day=15).bounds(1984, 12, 1, direction=direction),
cf.M(day=15).bounds(1984, 12, 3, direction=direction),
cf.M(day=15).bounds(1984, 12, 15, direction=direction),
cf.M(day=15).bounds(1984, 12, 20, direction=direction),
],
[(cf.dt('1984-01-01'), cf.dt('1984-02-01')),
(cf.dt('1984-12-01'), cf.dt('1985-01-01')),
(cf.dt('1984-12-01'), cf.dt('1985-01-01')),
(cf.dt('1984-11-15'), cf.dt('1984-12-15')),
(cf.dt('1984-11-15'), cf.dt('1984-12-15')),
(cf.dt('1984-12-15'), cf.dt('1985-01-15')),
(cf.dt('1984-12-15'), cf.dt('1985-01-15')),
]):
if direction is False:
y = y[::-1]
self.assertTrue(x==y, "{}!={}".format(x, y))
for x, y in zip(
[cf.D().bounds(1984, 1, 1, direction=direction),
cf.D().bounds(1984, 12, 3, direction=direction),
cf.D(hour=15).bounds(1984, 12, 1, direction=direction),
cf.D(hour=15).bounds(1984, 12, 1, 12, direction=direction),
cf.D(hour=15).bounds(1984, 12, 1, 15, direction=direction),
cf.D(hour=15).bounds(1984, 12, 1, 20, direction=direction),
],
[(cf.dt('1984-01-01'), cf.dt('1984-01-02')),
(cf.dt('1984-12-03'), cf.dt('1984-12-04')),
(cf.dt('1984-11-30 15:00'), cf.dt('1984-12-01 15:00')),
(cf.dt('1984-11-30 15:00'), cf.dt('1984-12-01 15:00')),
(cf.dt('1984-12-01 15:00'), cf.dt('1984-12-02 15:00')),
(cf.dt('1984-12-01 15:00'), cf.dt('1984-12-02 15:00')),
]):
if direction is False:
y = y[::-1]
self.assertTrue(x==y, "{}!={}".format(x, y))
#--- End: def
#--- End: class
if __name__ == "__main__":
print 'cf-python version:', cf.__version__
print 'cf-python path:' , os.path.abspath(cf.__file__)
print''
unittest.main(verbosity=2)
|
