def test_read_srml_columns_exist():
data = srml.read_srml(srml_testfile)
assert 'ghi_0' in data.columns
assert 'ghi_0_flag' in data.columns
assert 'dni_1' in data.columns
assert 'dni_1_flag' in data.columns
assert '7008' in data.columns
assert '7008_flag' in data.columns
def test_read_srml_dt_index(url, year, month):
data = srml.read_srml(url)
start = pd.Timestamp('{:04d}{:02d}01 00:00'.format(year, month))
start = start.tz_localize('Etc/GMT+8')
end = pd.Timestamp('{:04d}{:02d}31 23:59'.format(year, month))
end = end.tz_localize('Etc/GMT+8')
assert data.index[0] == start
assert data.index[-1] == end
assert (data.index[59::60].minute == 59).all()
assert str(year) not in data.columns
def test_read_srml_month_from_solardat():
url = 'http://solardat.uoregon.edu/download/Archive/EUPO1801.txt'
file_data = srml.read_srml(url)
requested = srml.read_srml_month_from_solardat('EU', 2018, 1)
assert file_data.equals(requested)
def test_read_srml_nans_exist():
data = srml.read_srml(srml_testfile)
assert isnan(data['dni_0'][1119])
assert data['dni_0_flag'][1119] == 99
def test_read_srml_remote():
srml.read_srml('http://solardat.uoregon.edu/download/Archive/EUPO1801.txt')
def test_read_srml():
srml.read_srml(srml_testfile)
