11
11
test_dir = os .path .dirname (
12
12
os .path .abspath (inspect .getfile (inspect .currentframe ())))
13
13
midc_testfile = os .path .join (test_dir , '../data/midc_20181014.txt' )
14
- midc_network_testfile = 'https://midcdmz.nrel.gov/apps/data_api.pl?site=UAT&begin=20181018&end=20181019'
14
+ midc_network_testfile = ('https://midcdmz.nrel.gov/apps/data_api.pl'
15
+ '?site=UAT&begin=20181018&end=20181019' )
16
+
15
17
16
18
@pytest .mark .parametrize ('field_name,expected' , [
17
19
('Temperature @ 2m [deg C]' , 'temp_air_@_2m' ),
@@ -34,6 +36,7 @@ def test_midc_format_index():
34
36
assert data .index [0 ] == start
35
37
assert data .index [- 1 ] == end
36
38
39
+
37
40
def test_midc_format_index_raw ():
38
41
data = pd .read_csv (midc_network_testfile )
39
42
data = midc .format_index_raw (data )
@@ -44,15 +47,16 @@ def test_midc_format_index_raw():
44
47
assert data .index [0 ] == start
45
48
assert data .index [- 1 ] == end
46
49
50
+
47
51
def test_read_midc_var_mapping_as_arg ():
48
52
data = midc .read_midc (midc_testfile , variable_map = midc .VARIABLE_MAP )
49
53
assert 'ghi_PSP' in data .columns
50
54
assert 'temp_air_@_2m' in data .columns
51
55
assert 'temp_air_@_50m' in data .columns
52
56
53
57
54
- start_ts = pd .Timestamp ('20181018' )
55
- end_ts = pd .Timestamp ('20181019' )
56
58
@network
57
59
def test_read_midc_raw_data_from_nrel ():
60
+ start_ts = pd .Timestamp ('20181018' )
61
+ end_ts = pd .Timestamp ('20181019' )
58
62
midc .read_midc_raw_data_from_nrel ('UAT' , start_ts , end_ts )
0 commit comments