import pytest
from os import path
import netCDF4
from tests.TestUtils import TestUtils
from netCDF4 import Dataset
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 = 'P:\\metocean-data\\open\\ERA5\\data\\Global\\msl_p'

        dir_test_data = 'D:\\workspace\\SD_Toolbox\\trunk\\tests\\test_data\\msl_p'
        # filename = 'era5_Global_Hs_1980.nc'
        # filepath = dir_test_data + filename

        input_data = data_acquisition.InputData()
        input_data.coord_list = [(4.2, 2.4), (42, 2.4), (42, 24), (4.2, 24)]
        # 2. When
        extract_slp = extract_sea_level_pressure_era5.ExtractSeaLevelPressureERA5(input_data)
        dataset_list = extract_slp.subset_sea_level_pressure(dir_test_data, 1981, 1982)
        # 3. Then
        assert dataset_list is not None