Skip to content

Commit 1e9ee64

Browse files
committed
rename index with new tz in coerce pvgis tmy
1 parent 91c169e commit 1e9ee64

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

pvlib/iotools/pvgis.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -405,6 +405,7 @@ def _coerce_and_roll_tmy(pvgis_data, tz, year):
405405
np.roll(pvgis_data, tz, axis=0),
406406
columns=pvgis_data.columns,
407407
index=new_index)
408+
new_pvgis_data.index.name = f'time({tzname})'
408409
return new_pvgis_data
409410

410411

pvlib/tests/iotools/test_pvgis.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -441,6 +441,7 @@ def test_get_pvgis_tmy_coerce_year():
441441
"""test utc_offset and coerce_year work as expected"""
442442
base_case, _, _, _ = get_pvgis_tmy(45, 8) # Turin
443443
assert str(base_case.index.tz) == 'UTC'
444+
assert base_case.index.name == 'time(UTC)'
444445
noon_test_data = [
445446
base_case[base_case.index.month == m].iloc[12]
446447
for m in range(1, 13)]
@@ -452,6 +453,7 @@ def test_get_pvgis_tmy_coerce_year():
452453
dec31_midnight = pd.Timestamp('1990-12-31 23:00:00', tz=cet_name)
453454
assert pvgis_data.index[0] == jan1_midnight
454455
assert pvgis_data.index[-1] == dec31_midnight
456+
assert pvgis_data.index.name == f'time({cet_name})'
455457
# spot check rolled data matches original
456458
for m, test_case in enumerate(noon_test_data):
457459
expected = pvgis_data[pvgis_data.index.month == m+1].iloc[12+cet_tz]
@@ -464,6 +466,7 @@ def test_get_pvgis_tmy_coerce_year():
464466
dec31_midnight = pd.Timestamp(f'{test_yr}-12-31 23:00:00', tz=cet_name)
465467
assert pvgis_data.index[0] == jan1_midnight
466468
assert pvgis_data.index[-1] == dec31_midnight
469+
assert pvgis_data.index.name == f'time({cet_name})'
467470
for m, test_case in enumerate(noon_test_data):
468471
expected = pvgis_data[pvgis_data.index.month == m+1].iloc[12+cet_tz]
469472
assert all(test_case == expected)
@@ -473,6 +476,7 @@ def test_get_pvgis_tmy_coerce_year():
473476
dec31_midnight = pd.Timestamp(f'{test_yr}-12-31 23:00:00', tz='UTC')
474477
assert pvgis_data.index[0] == jan1_midnight
475478
assert pvgis_data.index[-1] == dec31_midnight
479+
assert pvgis_data.index.name == 'time(UTC)'
476480
for m, test_case in enumerate(noon_test_data):
477481
expected = pvgis_data[pvgis_data.index.month == m+1].iloc[12]
478482
assert all(test_case == expected)

0 commit comments

Comments
 (0)