def test_read_midc_raw_data_from_nrel(): start_ts = pd.Timestamp('20181018') end_ts = pd.Timestamp('20181019') var_map = midc.MIDC_VARIABLE_MAP['UAT'] data = midc.read_midc_raw_data_from_nrel('UAT', start_ts, end_ts, var_map) for k, v in var_map.items(): assert v in data.columns assert data.index.size == 2880
def test_read_midc_header_length_mismatch(mocker): mock_data = mocker.MagicMock() with MIDC_RAW_SHORT_HEADER_TESTFILE.open() as f: mock_data.text = f.read() mocker.patch('pvlib.iotools.midc.requests.get', return_value=mock_data) start = pd.Timestamp('2019-11-15T00:00:00-06:00') end = pd.Timestamp('2019-11-15T23:59:00-06:00') data = midc.read_midc_raw_data_from_nrel('', start, end) assert isinstance(data.index, pd.DatetimeIndex) assert data.index[0] == start assert data.index[-1] == end
def test_read_midc_raw_data_from_nrel(): start_ts = pd.Timestamp('20181018') end_ts = pd.Timestamp('20181019') data = midc.read_midc_raw_data_from_nrel('UAT', start_ts, end_ts) assert 'dni_Normal' in data.columns assert data.index.size == 2880