sup3r.utilities.pytest.helpers.make_collect_chunks

sup3r.utilities.pytest.helpers.make_collect_chunks#

make_collect_chunks(td, ext='h5')[source]#

Make fake chunked output files for collection tests.

Parameters:

  • td (tempfile.TemporaryDirectory) – Test TemporaryDirectory

  • ext (str) – File extension for output files. Either ‘h5’ or ‘nc’. Default is ‘h5’.

Returns:

  • out_files (list) – List of filepaths to chunked files.

  • data (ndarray) – (spatial_1, spatial_2, temporal, features) High resolution forward pass output

  • ws_true (ndarray) – Windspeed between 0 and 20 in shape (spatial_1, spatial_2, temporal, 1)

  • wd_true (ndarray) – Windir between 0 and 360 in shape (spatial_1, spatial_2, temporal, 1)

  • features (list) – List of feature names corresponding to the last dimension of data [‘windspeed_100m’, ‘winddirection_100m’]

  • hr_lat_lon (ndarray) – Array of lat/lon for hr data. (spatial_1, spatial_2, 2) Last dimension has ordering (lat, lon)

  • hr_times (list) – List of np.datetime64 objects for hr data.

Contents