FIX: annual frequency indicators which have another axis with an 'A' label

Author: gdementen

larray_eurostat/tests/test_tsv.py

@@ -18,3 +18,14 @@ def test_eurostat_get_bad_dataset():
 def test_year_only_axes_as_int_and_not_object():
     data = eurostat_get('avia_ec_enterp')
     assert data.time.dtype.kind == 'i'
+
+
+# Check that annual frequency does not break when there is another axis with an 'A' label
+# https://github.com/larray-project/larray_eurostat/issues/34
+# An example of indicator with this problem is 'hsw_ph3_01'.
+# Keeping this test commented because the data is ~34Mb, so using it as a unit test
+# is probably not a good idea.
+# def test_annual_freq_with_other_a_label():
+#     data = eurostat_get('hsw_ph3_01')
+#     assert 'freq' not in data.axes
+

larray_eurostat/tsv.py

@@ -51,7 +51,7 @@ def _get_one(indicator, *, drop_markers=True):
 
             # If only one frequency: subset and return without redundant freq Axis (compatibility old API)
             if len(la_data.freq) == 1:
-                return la_data[la_data.freq.labels[0]]
+                return la_data[la_data.freq.i[0]]
             else:
                 return la_data