Coverage for src/susi/utils/header_checks.py: 100%

1 

2from ..io import FitsBatch 

3from ..base import IllegalStateException, Logging 

4from ..base.header_keys import SPATIAL_BIN 

5 

6log = Logging.get_logger() 

7 

8 

9def check_same_binning(batch: FitsBatch) -> None: 

10 bins = set([e['header'][SPATIAL_BIN] if SPATIAL_BIN in e['header'] else None for e in batch.batch]) 

11 if len(bins) > 1: 

12 log.critical("Different spacial binning applied to the frames loaded. Cannot continue. \n%s", bins) 

13 raise IllegalStateException('Data was spatially binned differently, see log for details')