fitscube 2.3.0__tar.gz → 2.3.2__tar.gz

{fitscube-2.3.0 → fitscube-2.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fitscube
-Version: 2.3.0
+Version: 2.3.2
 Summary: A package to produce produce FITS cubes.
 Project-URL: Homepage, https://github.com/AlecThomson/fitscube
 Project-URL: Bug Tracker, https://github.com/AlecThomson/fitscube/issues
@@ -58,6 +58,7 @@ Requires-Dist: sphinx-autodoc-typehints; extra == 'docs'
 Requires-Dist: sphinx-copybutton; extra == 'docs'
 Requires-Dist: sphinx>=7.0; extra == 'docs'
 Provides-Extra: test
+Requires-Dist: pytest-asyncio; extra == 'test'
 Requires-Dist: pytest-cov>=3; extra == 'test'
 Requires-Dist: pytest>=6; extra == 'test'
 Provides-Extra: uvloop

fitscube-2.3.2/fitscube/_version.py

@@ -0,0 +1,24 @@
+# file generated by vcs-versioning
+# don't change, don't track in version control
+from __future__ import annotations
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+version: str
+__version__: str
+__version_tuple__: tuple[int | str, ...]
+version_tuple: tuple[int | str, ...]
+commit_id: str | None
+__commit_id__: str | None
+
+__version__ = version = '2.3.2'
+__version_tuple__ = version_tuple = (2, 3, 2)
+
+__commit_id__ = commit_id = None

{fitscube-2.3.0 → fitscube-2.3.2}/fitscube/bounding_box.py

@@ -15,16 +15,17 @@ from fitscube.logging import logger
 
 @dataclass(frozen=True)
 class BoundingBox:
-    """Simple container to represent a bounding box"""
+    """Simple container to represent a bounding box. Maximum values can be
+    used as is when slicing."""
 
     xmin: int
     """Minimum x pixel"""
     xmax: int
     """Maximum x pixel"""
     ymin: int
-    """Minimum y pixel"""
+    """Minimum y pixel. Can be used as is in slice (e.g. is exclusive). """
     ymax: int
-    """Maximum y pixel"""
+    """Maximum y pixel Can be used as is in slice (e.g. is exclusive)."""
     original_shape: tuple[int, int]
     """The original shape of the image. If constructed against a cube this is the shape of a single plane."""
     y_span: int
@@ -62,6 +63,9 @@ def create_bound_box_plane(image_data: np.ndarray) -> BoundingBox | None:
     xmin, xmax = np.where(x_valid)[0][[0, -1]]
     ymin, ymax = np.where(y_valid)[0][[0, -1]]
 
+    xmax += 1
+    ymax += 1
+
     y_span = ymax - ymin
     x_span = xmax - xmin
 

{fitscube-2.3.0 → fitscube-2.3.2}/fitscube/combine_fits.py

@@ -16,10 +16,7 @@ import asyncio
 import warnings
 from io import BufferedRandom
 from pathlib import Path
-from typing import (
-    NamedTuple,
-    TypeVar,
-)
+from typing import Literal, NamedTuple, TypeVar
 
 import astropy.units as u
 import numpy as np
@@ -50,6 +47,8 @@ BIT_DICT = {
     16: 2,
     8: 1,
 }
+FLOAT_LENGTH = Literal[16, 32, 64]
+FLOAT_TYPE = {64: ">f8", 32: ">f4"}
 
 warnings.filterwarnings("ignore", category=UserWarning, module="astropy.io.fits")
 warnings.filterwarnings("ignore", category=VerifyWarning)
@@ -133,6 +132,41 @@ def isin_close(
     return np.isclose(element[:, None], test_element, atol, rtol).any(1)
 
 
+def grid_step(diffs: ArrayLike) -> float:
+    """Recover the fundamental channel step from the observed diffs.
+
+    On a regular grid with missing channels every diff is an integer multiple
+    of the channel width, so the gcd of the diffs recovers that width even when
+    no two surviving channels are adjacent -- the case where ``min(diffs)``
+    overestimates the step (e.g. present channels 0, 3, 5 give ``min == 2`` but
+    the true step is 1). Falls back to ``min(diffs)`` if the gcd is numerically
+    unstable or does not divide every diff, so the common case is unchanged (any
+    surviving adjacent pair makes the gcd equal the minimum diff exactly).
+
+    Args:
+        diffs (ArrayLike): Differences between consecutive spec values.
+
+    Returns:
+        float: Estimated regular grid step.
+    """
+    diffs = np.abs(np.asarray(diffs, dtype=np.longdouble))
+    min_diff = np.min(diffs)
+    tol = np.max(diffs) * 1e-6
+
+    g = diffs[0]
+    for d in diffs[1:]:
+        a, b = (g, d) if g >= d else (d, g)
+        # Tolerant Euclid: stop once the remainder is within the noise floor.
+        while b > tol:
+            a, b = b, a % b
+        g = a
+
+    divides_all = np.all(np.abs(np.round(diffs / g) - diffs / g) <= 1e-3)
+    if g <= tol or not divides_all:
+        return float(min_diff)
+    return float(g)
+
+
 def even_spacing(specs: u.Quantity, time_domain_mode: bool = False) -> SpequencyInfo:
     """Make the frequencies or times evenly spaced.
 
@@ -144,9 +178,9 @@ def even_spacing(specs: u.Quantity, time_domain_mode: bool = False) -> Spequency
     """
     specs_arr = specs.value.astype(np.longdouble)
     diffs = np.diff(specs_arr)
-    min_diff: float = np.min(diffs)
-    # Create a new array with the minimum difference
-    new_specs = np_arange_fix(specs_arr[0], specs_arr[-1], min_diff)
+    step = grid_step(diffs)
+    # Create a new array with the fundamental grid step
+    new_specs = np_arange_fix(specs_arr[0], specs_arr[-1], step)
     missing_chan_idx = np.logical_not(
         isin_close(new_specs, specs_arr, time_domain_mode)
     )
@@ -247,19 +281,26 @@ async def create_output_cube_coro(
     overwrite: bool = False,
     time_domain_mode: bool = False,
     bounding_box: BoundingBox | None = None,
+    float_length: FLOAT_LENGTH | None = None,
 ) -> InitResult:
-    """Initialize the data cube.
+    """Generate the output header and write a dummy cube to disk based on properties of the
+    inpute data. The output cube written here has a correctly formed header and a pre-zerod
+    data cube written as output.
 
     Args:
-        old_name (str): Old FITS file name
-        n_chan (int): Number of channels
-
-    Raises:
-        KeyError: If 2D and REFFREQ is not in header
-        ValueError: If not 2D and FREQ is not in header
+        old_name (Path): The path to a representative image to draw the base fits header from
+        out_cube (Path): Path of the output cube to create
+        specs (u.Quantity): Specification of the unit that denotes the 'cube' axis
+        ignore_spec (bool, optional): Whether the provided `specs` axis should be ignored. If True dummy placeholder fields added. Defaults to False.
+        has_beams (bool, optional): Indicates whether a CASA Beam table will also be generated and added to the output cube. Defaults to False.
+        single_beam (bool, optional): Indicates whether a constant restoring beam has been used among all input images. If so only the beam fields are need. Defaults to False.
+        overwrite (bool, optional): If True the out the output cube will overwrite any existing file. Defaults to False.
+        time_domain_mode (bool, optional): Whether to join images via the DATE-OBS (e.g. time) axis. If False cube joined along frequency axis. Defaults to False.
+        bounding_box (BoundingBox | None, optional): Whether a trimming operation should be applied to input images. If a BoundingBox is supplied this will be used to update the reference pixel position and data shape indicators. Defaults to None.
+        float_length (Literal[16, 32, 64] | None, optional): The precision of the output data. If None drawn from input data. Otherwise values accepted are 16, 32 and 64. Defaults to None.
 
     Returns:
-        InitResult: header, spec_idx, spec_fits_idx, is_2d
+        InitResult: Details of the output cube, including the output header
     """
 
     # define units if in time or freq domain
@@ -380,6 +421,23 @@ async def create_output_cube_coro(
     else:
         cube_shape[idx] = n_chan
 
+    logger.critical(f"{float_length=} {new_header['BITPIX']=}")
+    if float_length is not None:
+        # Per astropy docs
+        # bITPIX    numpy data type
+        # 8         numpy.uint8 (note it is UNsigned integer)
+        # 16        numpy.int16
+        # 32        numpy.int32
+        # 64        numpy.int64
+        # -32       numpy.float32
+        # -64       numpy.float64
+        assert float_length in list(BIT_DICT.keys()), (
+            f"{float_length=} not in {BIT_DICT=}"
+        )
+        bit_pix = int(float_length)
+        logger.info(f"Specified {float_length=}, corresponding to {bit_pix=}")
+        new_header["BITPIX"] = -abs(bit_pix)
+
     output_header = await create_cube_from_scratch_coro(
         output_file=out_cube, output_header=new_header, overwrite=overwrite
     )
@@ -657,6 +715,11 @@ async def process_channel(
     if invalidate_zeros:
         plane[plane == 0.0] = np.nan
 
+    if "BITPIX" in new_header:
+        bit_pix = abs(new_header["BITPIX"])
+        float_type = FLOAT_TYPE[bit_pix]
+        plane = plane.astype(float_type)
+
     await write_channel_to_cube_coro(
         file_handle=file_handle,
         plane=plane,
@@ -678,6 +741,7 @@ async def combine_fits_coro(
     time_domain_mode: bool = False,
     bounding_box: bool = False,
     invalidate_zeros: bool = False,
+    float_length: FLOAT_LENGTH | None = None,
 ) -> u.Quantity:
     """Combine FITS files into a cube.
     Can handle either frequency or time dimensions agnostically
@@ -690,6 +754,7 @@ async def combine_fits_coro(
         time_domain_mode (bool, optional): Work in time domain mode - make a time-cube. Default = False.
         bounding_box (bool, optional): Clip invalid/padded pixels when crafting the fits cube. When True an extra read of the input daata is needed, but output cube is smaller. Defaults to False.
         invalidate_zeros (bool, optionals): Set pixels whose values are exactly zero to NaNs. Defaults to False.
+        float_length (Literal[16, 32, 64] | None, optional): The floating point precision in bits to use when creating the output cube. If None the size of the input data are used. Defaults to None.
 
     Returns:
         tuple[fits.HDUList, u.Quantity]: The combined FITS cube and frequencies
@@ -760,6 +825,7 @@ async def combine_fits_coro(
         overwrite=overwrite,
         time_domain_mode=time_domain_mode,
         bounding_box=final_bounding_box,
+        float_length=float_length,
     )
 
     new_channels = np.arange(len(specs))
@@ -881,6 +947,13 @@ def get_parser(
         action="store_true",
         help="Set pixels whose values are exactly zero to NaNs",
     )
+    parser.add_argument(
+        "--floating",
+        type=int,
+        choices=(8, 16, 32, 64),
+        default=None,
+        help="The number of floating point bits to use in the out cube. If None the input data precision is used.",
+    )
 
     return parser
 
@@ -923,6 +996,7 @@ def cli(args: argparse.Namespace | None = None) -> None:
         time_domain_mode=time_domain_mode,
         bounding_box=args.bounding_box,
         invalidate_zeros=args.invalidate_zeros,
+        float_length=args.floating,
     )
 
     spequency = "times" if time_domain_mode else "frequencies"

{fitscube-2.3.0 → fitscube-2.3.2}/pyproject.toml

@@ -38,6 +38,7 @@ dependencies = [
 [project.optional-dependencies]
 test = [
   "pytest >=6",
+  "pytest-asyncio",
   "pytest-cov >=3",
 ]
 dev = [
@@ -82,6 +83,7 @@ addopts = ["-ra", "--showlocals", "--strict-markers", "--strict-config"]
 xfail_strict = true
 filterwarnings = [
   "error",
+  "ignore:The TestRunner",
 ]
 log_cli_level = "INFO"
 testpaths = [

{fitscube-2.3.0 → fitscube-2.3.2}/tests/conftest.py

@@ -4,6 +4,7 @@ from pathlib import Path
 from shutil import unpack_archive
 
 import pytest
+from astropy.io import fits
 
 EXAMPLE_HEADER = "SIMPLE  =                    T / conforms to FITS standard                      BITPIX  =                  -32 / array data type                                NAXIS   =                    4 / number of array dimensions                     NAXIS1  =                 6192                                                  NAXIS2  =                 6192                                                  NAXIS3  =                    1                                                  NAXIS4  =                   72                                                  EXTEND  =                    T                                                  BSCALE  =                  1.0                                                  BZERO   =                  0.0                                                  BUNIT   = 'JY/BEAM '                                                            BMAJ    =  0.00398184964207042                                                  BMIN    =  0.00332268385509243                                                  BPA     =     77.3858939868987                                                  EQUINOX =               2000.0                                                  LONPOLE =                180.0                                                  BTYPE   = 'Intensity'                                                           TELESCOP= 'ASKAP   '                                                            OBJECT  = 'EMU_1141-55'                                                         ORIGIN  = 'WSClean '                                                            CTYPE1  = 'RA---SIN'                                                            CRPIX1  =               3097.0                                                  CRVAL1  =     173.522555019647                                                  CDELT1  =            -0.000625                                                  CUNIT1  = 'deg     '                                                            CTYPE2  = 'DEC--SIN'                                                            CRPIX2  =               3097.0                                                  CRVAL2  =    -55.3190947400628                                                  CDELT2  =             0.000625                                                  CUNIT2  = 'deg     '                                                            CTYPE3  = 'STOKES  '                                                            CRPIX3  =                  1.0                                                  CRVAL3  =                  1.0                                                  CDELT3  =                  1.0                                                  CUNIT3  = ''                                                                    CTYPE4  = 'FREQ    '                                                            CRPIX4  =                    1                                                  CRVAL4  =     801490740.740741                                                  CDELT4  =            4000000.0                                                  CUNIT4  = 'Hz      '                                                            SPECSYS = 'TOPOCENT'                                                            DATE-OBS= '2023-01-08T15:36:40.9'                                               WSCDATAC= 'DATA    '                                                            WSCVDATE= '2022-10-21'                                                          WSCVERSI= '3.2     '                                                            WSCWEIGH= 'Briggs''(-0)'                                                        WSCENVIS=     1026544.24656423                                                  WSCFIELD=                  0.0                                                  WSCGAIN =                  0.1                                                  WSCGKRNL=                  7.0                                                  WSCIMGWG=     4052355.35920529                                                  WSCMAJOR=                  9.0                                                  WSCMGAIN=                  0.9                                                  WSCMINOR=              62860.0                                                  WSCNEGCM=                  1.0                                                  WSCNEGST=                  0.0                                                  WSCNITER=            5000000.0                                                  WSCNORMF=     4052355.35920529                                                  WSCNVIS =            7648518.0                                                  WSCNWLAY=                  1.0                                                  WSCTHRES=                  0.0                                                  WSCVWSUM=           30594072.0                                                  COMMENT   FITS (Flexible Image Transport System) format is defined in 'AstronomyCOMMENT   and Astrophysics', volume 376, page 359; bibcode: 2001A&A...376..359H HISTORY wsclean -abs-mem 200 -local-rms-window 20 -size 6192 6192 -local-rms -foHISTORY rce-mask-rounds 4 -auto-mask 8.0 -auto-threshold 3.0 -channels-out 72 -mHISTORY gain 0.9 -nmiter 10 -niter 5000000 -multiscale-scale-bias 0.8 -multiscalHISTORY e-scales 0,4,8,16,24,32,48,64,92,128,196,512,796,1025 -fit-spectral-pol HISTORY 4 -weight briggs -0 -data-column DATA -scale 2.25asec -gridder wgridder HISTORY -wgridder-accuracy 0.0001 -join-channels -minuv-l 50.0 -beam-fitting-sizHISTORY e 1.25 -deconvolution-channels 8 -parallel-gridding 18 -temp-dir /dev/shHISTORY m/gal16b.8974844 -pol i -save-source-list -name /dev/shm/gal16b.8974844/HISTORY SB47138.EMU_1141-55.beam19.i /scratch3/gal16b/emu_download/flint_jollytrHISTORY actor/47138/SB47138.EMU_1141-55.beam19.ms                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                               END                                                                             "
 
@@ -65,3 +66,31 @@ def time_image_paths(tmpdir) -> list[Path]:
     image_paths.sort()
 
     return image_paths
+
+
+@pytest.fixture
+def time_image_paths_withzeroborder(tmpdir) -> list[Path]:
+    """Same as above but zero out border pixels in the data"""
+    tmp_dir = Path(tmpdir) / "time_images"
+    tmp_dir.mkdir(exist_ok=True, parents=True)
+    images_zip = Path(__file__).parent / "data" / "time_images.zip"
+
+    unpack_archive(images_zip, tmp_dir)
+    image_paths = list(tmp_dir.glob("*fits"))
+    image_paths.sort()
+
+    output_paths = []
+    for image in image_paths:
+        with fits.open(image, memmap=False, lazy_load_hdus=False) as a:
+            data = a[0].data
+            data[..., :5, :] = 0
+            data[..., -5:, :] = 0
+            data[..., :, :5] = 0
+            data[..., :, -5:] = 0
+
+            a[0].data = data
+            image_output = image.with_suffix(".zero.fits")
+            fits.writeto(image_output, data=data, header=a[0].header)
+            output_paths.append(image_output)
+
+    return output_paths

fitscube-2.3.2/tests/test_bb.py

@@ -0,0 +1,97 @@
+# Tests to ensure the bounding box functionality is working
+from __future__ import annotations
+
+import numpy as np
+import pytest
+from fitscube.bounding_box import (
+    BoundingBox,
+    create_bound_box_plane,
+    extract_common_bounding_box,
+)
+
+
+def test_bound_box_plane() -> None:
+    """See if we can make the bounding box correctly"""
+
+    image = np.zeros((100, 100))
+
+    bb = create_bound_box_plane(image_data=image)
+    assert isinstance(bb, BoundingBox)
+    assert bb.xmin == 0
+    assert bb.ymin == 0
+    assert bb.xmax == 100
+    assert bb.ymax == 100
+    assert bb.x_span == 100
+    assert bb.y_span == 100
+
+    image[:, :4] = np.nan
+    image[96:, :] = np.nan
+
+    bb = create_bound_box_plane(image_data=image)
+    assert isinstance(bb, BoundingBox)
+    assert bb.xmin == 0
+    assert bb.ymin == 4
+    assert bb.xmax == 96
+    assert bb.ymax == 100
+    assert bb.x_span == 96
+    assert bb.y_span == 96
+
+
+def test_extract_common_bounding_box() -> None:
+    """Construct a set of bounding boxes to ensure a largest fits all
+    one can be created"""
+
+    bbs = []
+    for i in range(2, 8, 1):
+        image = np.zeros((100, 100))
+
+        print(i)
+
+        image[:i, :] = np.nan
+        image[:, -i:] = np.nan
+        assert np.sum(np.isnan(image)) > 1
+        bbs.append(create_bound_box_plane(image_data=image))
+
+    assert len(bbs) == 6
+
+    bb = extract_common_bounding_box(bounding_boxes=bbs)
+    assert bb.xmin == 2
+    assert bb.ymin == 0
+    assert bb.xmax == 100
+    assert bb.ymax == 98
+
+
+def test_extract_common_bounding_box_error() -> None:
+    """See if the correct errors are raised"""
+
+    with pytest.raises(ValueError, match="No valid"):
+        extract_common_bounding_box(bounding_boxes=[None, None, None])
+
+    bbs = []
+    for i in range(2, 8, 1):
+        image = np.zeros((100 + i, 100))
+        bbs.append(create_bound_box_plane(image_data=image))
+
+    with pytest.raises(ValueError, match="Different shapes"):
+        extract_common_bounding_box(bounding_boxes=bbs)
+
+
+# @pytest.mark.asyncio
+# async def test_get_bounding_box_from_fits(time_image_paths) -> None:
+#     """!!! NOTE: Sometimes this test can raise the following error:
+
+#     > Exception ignored in: <socket.socket fd=-1, family=1, type=1, proto=0>
+
+#     Not sure why it is capable of such a thing.
+#     """
+#     futures = [
+#         await get_bounding_box_for_fits_coro(fits_path=fits_path)
+#         for fits_path in time_image_paths
+#     ]
+#     assert all(isinstance(bb, BoundingBox) for bb in futures)
+
+#     common_bb = extract_common_bounding_box(bounding_boxes=futures)
+#     assert common_bb.xmin == 0
+#     assert common_bb.ymin == 0
+#     assert common_bb.xmax == 100
+#     assert common_bb.ymax == 100

{fitscube-2.3.0 → fitscube-2.3.2}/tests/test_frequencies.py

@@ -6,7 +6,7 @@ import astropy.units as u
 import numpy as np
 import pytest
 from astropy.io import fits
-from fitscube.combine_fits import combine_fits, parse_specs
+from fitscube.combine_fits import combine_fits, even_spacing, parse_specs
 
 
 @pytest.fixture
@@ -108,3 +108,16 @@ def test_uneven_combine(
             assert chan in (1, 2)
             continue
         assert np.allclose(plane, image)
+
+
+def test_even_spacing_non_adjacent_gaps():
+    # Present channels 0, 3, 5 GHz on a 1 GHz grid (1, 2, 4 missing). No two
+    # surviving channels are adjacent, so min(diffs) == 2 GHz would mis-grid the
+    # axis to [0, 2, 4] and drop the real channels. The gcd of the diffs
+    # [3, 2] GHz recovers the true 1 GHz step.
+    specs = np.array([0.0, 3.0, 5.0]) * u.GHz
+    new_specs, missing_chan_idx = even_spacing(specs)
+    assert len(new_specs) == 6, "should rebuild the full 0..5 GHz grid"
+    assert missing_chan_idx.sum() == 3
+    expected_missing = np.array([False, True, True, False, True, False])
+    assert np.array_equal(missing_chan_idx, expected_missing)

{fitscube-2.3.0 → fitscube-2.3.2}/tests/test_times.py

@@ -92,6 +92,14 @@ def test_even_combine(file_list: list[Path], even_specs: u.Quantity, output_file
 def test_uneven_combine(
     file_list: list[Path], even_specs: u.Quantity, output_file: Path
 ):
+    """!!!! NOTE: For some unknown reason thios test *SOMETIMES* raises the
+    following error:
+
+    > FAILED tests/test_times.py::test_uneven - ValueError: operands could not be broadcast together with shapes (5,) (6,)
+
+    Completely unclear to me why
+    """
+
     # uneven_specs = np.concatenate([even_specs[0:1], even_specs[3:]])
     file_array = np.array(file_list)
     uneven_files = np.concatenate([file_array[0:1], file_array[3:]]).tolist()
@@ -149,3 +157,113 @@ def test_wsclean_images_create_axis(time_image_paths, tmpdir) -> None:
         # The TIME axis will be appended as a new dimension
         cube_image_data = cube_data[i]
         assert np.allclose(image_data.squeeze(), cube_image_data.squeeze())
+
+
+@pytest.mark.filterwarnings("ignore:'datfix' made the change")
+def test_wsclean_images_create_axis_floating(time_image_paths, tmpdir) -> None:
+    """Ensure that the combined cube conforms to the input data"""
+
+    tmpdir = Path(tmpdir) / "time_cube_combine"
+    tmpdir.mkdir(parents=True, exist_ok=True)
+    out_cube = tmpdir / "time_cube_mate.fits"
+
+    for float_length, float_rep in ((32, ">f4"), (64, ">f8")):
+        combine_fits(
+            file_list=time_image_paths,
+            out_cube=out_cube,
+            overwrite=True,
+            time_domain_mode=True,
+            float_length=float_length,  # type: ignore [arg-type]
+        )
+
+        cube_data = fits.getdata(out_cube)
+        assert cube_data.dtype == float_rep
+        for i, time_image_path in enumerate(time_image_paths):
+            image_data = fits.getdata(time_image_path)
+            # The TIME axis will be appended as a new dimension
+            cube_image_data = cube_data[i]
+            assert np.allclose(image_data.squeeze(), cube_image_data.squeeze())
+
+
+@pytest.mark.filterwarnings("ignore:'datfix' made the change")
+def test_wsclean_images_create_axis_withbb(time_image_paths, tmpdir) -> None:
+    """Ensure that the combined cube conforms to the input data. This will use the
+    bounding box option. This should led to no change."""
+
+    tmpdir = Path(tmpdir) / "time_cube_combine"
+    tmpdir.mkdir(parents=True, exist_ok=True)
+    out_cube = tmpdir / "time_cube_mate.fits"
+
+    combine_fits(
+        file_list=time_image_paths,
+        out_cube=out_cube,
+        overwrite=True,
+        time_domain_mode=True,
+        bounding_box=True,
+    )
+
+    cube_data = fits.getdata(out_cube)
+    for i, time_image_path in enumerate(time_image_paths):
+        image_data = fits.getdata(time_image_path)
+        # The TIME axis will be appended as a new dimension
+        cube_image_data = cube_data[i]
+        assert np.allclose(image_data.squeeze(), cube_image_data.squeeze())
+
+
+@pytest.mark.filterwarnings("ignore:'datfix' made the change")
+def test_wsclean_images_create_axis_withbb_noinvalidate0(
+    time_image_paths_withzeroborder, tmpdir
+) -> None:
+    """Ensure that the combined cube conforms to the input data. This will use the
+    bounding box option. This should led to no change."""
+
+    tmpdir = Path(tmpdir) / "time_cube_combine_zeros"
+    tmpdir.mkdir(parents=True, exist_ok=True)
+    out_cube = tmpdir / "time_cube_mate.fits"
+
+    combine_fits(
+        file_list=time_image_paths_withzeroborder,
+        out_cube=out_cube,
+        overwrite=True,
+        time_domain_mode=True,
+        bounding_box=True,
+        invalidate_zeros=False,
+    )
+
+    cube_data = fits.getdata(out_cube)
+    assert cube_data.squeeze().shape[-2:] == (100, 100)
+    for i, time_image_path in enumerate(time_image_paths_withzeroborder):
+        image_data = fits.getdata(time_image_path)
+        # The TIME axis will be appended as a new dimension
+        cube_image_data = cube_data[i]
+        assert np.allclose(image_data.squeeze(), cube_image_data.squeeze())
+
+
+@pytest.mark.filterwarnings("ignore:'datfix' made the change")
+def test_wsclean_images_create_axis_withbb_invalidatezeros(
+    time_image_paths_withzeroborder, tmpdir
+) -> None:
+    """Ensure that the combined cube conforms to the input data. This will use the
+    bounding box option. This should led to no change. Here we also get fitscube to
+    mark pixels that are exactly 0 to be nans"""
+
+    tmpdir = Path(tmpdir) / "time_cube_combine_zeros"
+    tmpdir.mkdir(parents=True, exist_ok=True)
+    out_cube = tmpdir / "time_cube_mate.fits"
+
+    combine_fits(
+        file_list=time_image_paths_withzeroborder,
+        out_cube=out_cube,
+        overwrite=True,
+        time_domain_mode=True,
+        bounding_box=True,
+        invalidate_zeros=True,
+    )
+
+    cube_data = fits.getdata(out_cube)
+    assert cube_data.squeeze().shape[-2:] == (90, 90)
+    for i, time_image_path in enumerate(time_image_paths_withzeroborder):
+        image_data = fits.getdata(time_image_path)
+        # The TIME axis will be appended as a new dimension
+        cube_image_data = cube_data[i]
+        assert np.allclose(image_data.squeeze()[5:-5, 5:-5], cube_image_data.squeeze())

fitscube-2.3.0/fitscube/_version.py

@@ -1,34 +0,0 @@
-# file generated by setuptools-scm
-# don't change, don't track in version control
-
-__all__ = [
-    "__version__",
-    "__version_tuple__",
-    "version",
-    "version_tuple",
-    "__commit_id__",
-    "commit_id",
-]
-
-TYPE_CHECKING = False
-if TYPE_CHECKING:
-    from typing import Tuple
-    from typing import Union
-
-    VERSION_TUPLE = Tuple[Union[int, str], ...]
-    COMMIT_ID = Union[str, None]
-else:
-    VERSION_TUPLE = object
-    COMMIT_ID = object
-
-version: str
-__version__: str
-__version_tuple__: VERSION_TUPLE
-version_tuple: VERSION_TUPLE
-commit_id: COMMIT_ID
-__commit_id__: COMMIT_ID
-
-__version__ = version = '2.3.0'
-__version_tuple__ = version_tuple = (2, 3, 0)
-
-__commit_id__ = commit_id = None