sup3r.utilities.pytest.helpers.make_fake_h5_chunks

sup3r.utilities.pytest.helpers.make_fake_h5_chunks#

make_fake_h5_chunks(td)[source]#

Make fake h5 chunked output files for a 5x spatial 2x temporal multi-node forward pass output.

Parameters:

td (tempfile.TemporaryDirectory) – Test TemporaryDirectory

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’]

  • t_slices_lr (list) – List of low res temporal slices

  • t_slices_hr (list) – List of high res temporal slices

  • s_slices_lr (list) – List of low res spatial slices

  • s_slices_hr (list) – List of high res spatial slices

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

  • low_res_times (list) – List of np.datetime64 objects for coarse data.