import unittest
import pandas as pd
from pyinterpolate.pipelines.samples import download_air_quality_poland


# TODO: this test is very slow, it must be refactored or controlled
# class TestDownloadAirQualityDataPoland(unittest.TestCase):
#
#     def test_download_air_quality_data_poland(self):
#         for sample in ['CO', 'SO2', 'PM2.5', 'PM10', 'NO2', 'O3', 'C6H6']:
#             df = download_air_quality_poland(sample)
#             self.assertIsInstance(df, pd.DataFrame)
#             self.assertEqual(
#                 {'station_id', 'x', 'y', sample},
#                 set(df.columns)
#             )
#             self.assertGreater(df[sample].mean(), 0)
