import pytest
import os
from os import path
import netCDF4
from tests.TestUtils import TestUtils
from netCDF4 import Dataset

from SDToolBox import main as main
from SDToolBox.extract_data_EARTH import ExtractDataEARTH
from SDToolBox.input_data import InputData
from SDToolBox import output_messages as om


class TestExtractDataEARTH:

    @pytest.mark.systemtest
    def test_given_EARTH_folder_then_subset_collection_is_extracted_with_input_coordinates(self):
        # 1. Given
        # When using local data you can just replace the comment in these lines
        dir_test_data = TestUtils.get_local_test_data_dir('meteo_1_Global')
        # dir_test_data = 'P:\11200665-c3s-codec\2_Hydro\DMI_meteo\meteo_1_Global\RCP45_'

        input_data = InputData()
        input_data.input_EARTH_scenario = 'RCP45'
        input_data.input_variables = ['var151']
        input_data.input_years = [2041, 2042]
        input_data.input_coordinates = \
            [(4.2, 2.4), (42, 2.4), (42, 24), (4.2, 24)]

        # 2. When
        extract_data = ExtractDataEARTH(input_data)
        output_data = extract_data.subset_sea_level_pressure(dir_test_data)

        # 3. Then
        assert output_data is not None
        data_dict = output_data.data_dict
        assert data_dict is not None
        assert data_dict[output_data.var_time_key] is not None
        assert data_dict[output_data.var_lon_key] is not None
        assert data_dict[output_data.var_lat_key] is not None
        assert data_dict[output_data.var_val_key] is not None
        assert data_dict[output_data.var_val_key]['var151'] is not None