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

from SDToolBox import main as main
from SDToolBox import extract_sea_level_pressure_era5
from SDToolBox import data_acquisition
import numpy as np


class Test_subset_spl_era_5:

    @pytest.mark.systemtest
    def test_slp_folder_then_subset_collection_is_extracted(self):
        # 1. Given
        # When using local data you can just replace the comment in these lines
        #dir_test_data = TestUtils.get_test_data_dir('netCDF_Waves_data')
        #dir_test_data = 'P:\\metocean-data\\open\\ERA5\\data\\Global'

        area_latitude = [45.0,33.0]
        area_longitude = [127.5,141.75]
        dir_test_data = 'D:\\workspace\\SD_Toolbox\\trunk\\tests\\test_data\\' # TODO to be removed
        
        input_data = data_acquisition.InputData()
        input_data.coord_list = [(4.2, 2.4)]

        # 2. When
        extract_slp = extract_slp.ExtractSeaLevelPressureERA5(input_data)

        # not sure about some parameters

        dataset_list = extract_slp.subset_slp_era5(self, area_latitude, area_longitude, 1979, 2016, dir_test_data, 1, 1, 1, 1)

        # 3. Then
        assert dataset_list is not None