ftag.mock#

Functions#

get_mock_file(→ tuple[str, h5py.File])

Get a mock file for testing.

Module Contents#

ftag.mock.get_mock_file(num_jets: int = 1000, fname: str | None = None, tracks_name: str = 'tracks', num_tracks: int = 40) → tuple[str, h5py.File]#

Get a mock file for testing.

Parameters:

  • num_jets (int, optional) – Number of jets in the file, by default 1000

  • fname (str | None, optional) – Name of the file, by default None

  • tracks_name (str, optional) – Name of the tracks dataset, by default “tracks”

  • num_tracks (int, optional) – Number of tracks per jet, by default 40

Returns:

Tuple with the path and the h5 file

Return type:

tuple[str, h5py.File]