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
|
import pandas
import ulmo
import test_util
test_sets = [
{
'filename': 'cpc/drought/palmer88',
'start': '1989-01-22',
'end': '1989-01-28',
'state': 'TX',
'climate_division': 1,
'values': [{
'cmi': -0.11,
'pdsi': 0.0,
'period': '1989-01-22/1989-01-28',
'potential_evap': 0.1,
'precipitation': 0.4,
'runoff': 0.0,
'soil_moisture_lower': 0.3,
'soil_moisture_upper': 0.3,
'temperature': 43.0
}],
},
{
'filename': 'cpc/drought/palmer94',
'start': '1995-02-19',
'end': '1995-02-25',
'state': 'NC',
'climate_division': 4,
'values': [{
'cmi': 1.92,
'pdsi': 0.88,
'period': '1995-02-19/1995-02-25',
'potential_evap': 0.17,
'precipitation': 0.1,
'runoff': 0.0,
'soil_moisture_lower': 5.0,
'soil_moisture_upper': 0.93,
'temperature': 47.0
}]
},
{
'filename': 'cpc/drought/palmer99',
'start': '1999-07-25',
'end': '1999-07-31',
'state': 'TX',
'climate_division': 3,
'values': [{
'cmi': -2.31,
'pdsi': -1.37,
'period': '1999-07-25/1999-07-31',
'potential_evap': 1.75,
'precipitation': 0.0,
'runoff': 0.0,
'soil_moisture_lower': 1.68,
'soil_moisture_upper': 0.0,
'temperature': 87.0
}]
},
{
'filename': 'cpc/drought/palmer10',
'start': '2010-5-20',
'end': '2010-6-13',
'state': 'AL',
'climate_division': 1,
'values': [{
'cmi': 0.05,
'pdsi': 0.53,
'period': '2010-06-06/2010-06-12',
'potential_evap': 1.37,
'precipitation': 0.71,
'runoff': 0.0,
'soil_moisture_lower': 4.86,
'soil_moisture_upper': 0.0,
'temperature': 77.6
}]
},
]
def test_get_data_by_state():
for test_set in test_sets:
with test_util.mocked_urls(test_set['filename']):
data = ulmo.cpc.drought.get_data(
state=test_set['state'], start=test_set['start'],
end=test_set['end'])
assert len(data) == 1
assert test_set['state'] in data
def test_get_data():
for test_set in test_sets:
with test_util.mocked_urls(test_set['filename']):
data = ulmo.cpc.drought.get_data(start=test_set['start'], end=test_set['end'])
values = data.get(test_set['state'], {}).get(test_set['climate_division'])
test_values = test_set['values']
for test_value in test_values:
assert test_value in values
def test_get_data_as_dataframe():
with test_util.mocked_urls('cpc/drought/palmer10'):
data = ulmo.cpc.drought.get_data(
start='2010-5-20', end='2010-6-13', as_dataframe=True)
assert isinstance(data, pandas.DataFrame)
# TODO: add proper marks to these
#def test_prelim_overlap_doesnt_break():
#data = ulmo.cpc.drought.get_data(start='2012-01-01', end='2013-03-07',
#as_dataframe=True)
#assert len(data) > 0
#def test_prelim_overlap_start_end_dates_obeyed():
#data = ulmo.cpc.drought.get_data(start='2012-01-01', end='2013-03-07',
#as_dataframe=True)
#assert data['period'][0] == pandas.Period('2012-01-01/2012-01-07', 'W-SAT')
#assert data['period'][data.index[-1]] == pandas.Period('2013-03-03/2013-03-09', 'W-SAT')
|
