From b155f31e30cbedec9182d1fe85864e17349dcf22 Mon Sep 17 00:00:00 2001
From: Jan Griesfeller <jang@met.no>
Date: Thu, 7 Nov 2024 16:41:54 +0100
Subject: [PATCH] WIP

---
 tests/test_PyerocomBinding.py | 79 +++++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)
 create mode 100644 tests/test_PyerocomBinding.py

diff --git a/tests/test_PyerocomBinding.py b/tests/test_PyerocomBinding.py
new file mode 100644
index 0000000..1f2db19
--- /dev/null
+++ b/tests/test_PyerocomBinding.py
@@ -0,0 +1,79 @@
+import unittest
+import urllib.request
+
+import pyaro
+import pyaro.timeseries
+from pyaro.timeseries.Wrappers import VariableNameChangingReader
+
+TEST_URL = "https://prod-actris-md.nilu.no/Version"
+VOCABULARY_URL = "https://prod-actris-md.nilu.no/V"
+
+
+class TestActrisEbasTimeSeriesReader(unittest.TestCase):
+    engine = "actrisebas"
+
+    station_filter = {
+        "stations": {
+            "include": ["Birkenes II", "Jungfraujoch", "Ispra", "Melpitz", "Westerland"]
+        },
+    }
+    # vars_to_read = ["ozone mass concentration"]
+    # vars_to_read = ["aerosol particle sulphate mass concentration"]
+    vars_to_read = ["aerosol particle elemental carbon mass concentration"]
+    # pyaerocom_vars_to_read = ["conco3"]
+    pyaerocom_vars_to_read = ["vmro3"]
+
+    # pyaerocom_vars_to_read = ["concso4t"]
+
+    # Try testing the pyaerocom binding if pyaerocom is installed
+    try:
+        from pyaerocom.io import ReadUngridded
+        pyaerocom_installed_flag = True
+    except ImportError:
+        pyaerocom_installed_flag = False
+        pass
+
+    AERONETVAR = "od440aer"
+
+
+    def test_pyaerocom_aeronet(self):
+        # test reading via pyaerocom
+        try:
+            from pyaerocom.io.pyaro.pyaro_config import PyaroConfig
+            from pyaerocom.io import ReadUngridded
+        except ImportError:
+            assert("pyaerocom not installed")
+            return
+
+        data_name = "aeronettest"
+        data_id = "aeronetsunreader"
+        url = "https://pyaerocom.met.no/pyaro-suppl/testdata/aeronetsun_testdata.csv"
+        obsconfig = PyaroConfig(
+            name=data_name,
+            data_id=data_id,
+            filename_or_obj_or_url=url,
+            filters={"variables": {"include": ["AOD_440nm"]}},
+            name_map={"AOD_440nm": self.AERONETVAR},
+        )
+        reader = ReadUngridded(f"{data_name}")
+        data = reader.read(vars_to_retrieve=self.AERONETVAR, configs=obsconfig)
+        self.assertGreaterEqual(len(data.unique_station_names), 4)
+        self.assertIn("Alta_Floresta", data.unique_station_names)
+
+
+    def test_api_reading_pyaerocom_naming(self):
+        # test access to the EBAS API
+        # test variable by variable
+        for _var in self.pyaerocom_vars_to_read:
+            engine = pyaro.list_timeseries_engines()[self.engine]
+            with engine.open(filters=self.station_filter, vars_to_read=[_var]) as ts:
+                self.assertGreaterEqual(len(ts.variables()), 1)
+                self.assertGreaterEqual(len(ts.stations()), 2)
+                self.assertGreaterEqual(len(ts._data[ts.variables()[0]]), 1000)
+                self.assertGreaterEqual(len(ts.data(ts.variables()[0])), 1000)
+                self.assertIn("revision", ts.metadata())
+
+
+
+if __name__ == "__main__":
+    unittest.main()