imap-processing 0.7.0__py3-none-any.whl → 0.9.0__py3-none-any.whl

imap_processing/tests/ialirt/unit/test_process_codicelo.py

@@ -0,0 +1,106 @@
+import numpy as np
+import pytest
+
+from imap_processing import imap_module_directory
+from imap_processing.cdf.utils import load_cdf
+from imap_processing.ialirt.l0.process_codicelo import (
+    append_cod_lo_data,
+    find_groups,
+    process_codicelo,
+)
+from imap_processing.utils import packet_file_to_datasets
+
+
+@pytest.fixture(scope="session")
+def xtce_codicelo_path():
+    """Returns the xtce directory."""
+    return (
+        imap_module_directory / "ialirt" / "packet_definitions" / "ialirt_codicelo.xml"
+    )
+
+
+@pytest.fixture(scope="session")
+def binary_packet_path():
+    """Returns the xtce directory."""
+    return (
+        imap_module_directory
+        / "tests"
+        / "ialirt"
+        / "test_data"
+        / "l0"
+        / "apid01152.tlm"
+    )
+
+
+@pytest.fixture(scope="session")
+def codicelo_validation_data():
+    """Returns the test data directory."""
+    data_path = (
+        imap_module_directory
+        / "tests"
+        / "ialirt"
+        / "test_data"
+        / "l0"
+        / "imap_codice_l1a_lo-ialirt_20241110193700_v0.0.0.cdf"
+    )
+    data = load_cdf(data_path)
+
+    return data
+
+
+@pytest.fixture()
+def codicelo_test_data(binary_packet_path, xtce_codicelo_path):
+    """Create xarray data"""
+    apid = 1152
+    codicelo_test_data = packet_file_to_datasets(
+        binary_packet_path, xtce_codicelo_path
+    )[apid]
+
+    return codicelo_test_data
+
+
+def test_find_groups(codicelo_test_data):
+    """Tests find_groups"""
+
+    grouped_data = find_groups(codicelo_test_data)
+    unique_groups = np.unique(grouped_data["group"])
+    for group in unique_groups:
+        group_data = grouped_data["cod_lo_counter"].values[
+            grouped_data["group"] == group
+        ]
+        np.testing.assert_array_equal(group_data, np.arange(233))
+
+
+def test_append_cod_lo_data(codicelo_test_data):
+    """Tests append_cod_lo_data"""
+
+    grouped_data = find_groups(codicelo_test_data)
+    unique_groups = np.unique(grouped_data["group"])
+    for group in unique_groups:
+        mask = grouped_data["group"] == group
+        filtered_indices = np.where(mask)[0]
+        group_data = grouped_data.isel(epoch=filtered_indices)
+        expected_cod_lo_counter = np.repeat(group_data["cod_lo_counter"].values, 15)
+        appended_data = append_cod_lo_data(group_data)
+        assert np.array_equal(
+            appended_data["cod_lo_counter"].values, expected_cod_lo_counter.astype(int)
+        )
+
+
+def test_process_codicelo(codicelo_test_data, codicelo_validation_data, caplog):
+    """Tests process_codicelo."""
+    codicelo_product = process_codicelo(codicelo_test_data)
+    assert codicelo_product == [{}]
+
+    indices = (codicelo_test_data["cod_lo_acq"] != 0).values.nonzero()[0]
+    codicelo_test_data["cod_lo_counter"].values[indices[0] : indices[0] + 233] = (
+        np.random.permutation(233)
+    )
+
+    with caplog.at_level("WARNING"):
+        process_codicelo(codicelo_test_data)
+
+    assert any(
+        "does not contain all values from 0 to 232 without duplicates" in message
+        for message in caplog.text.splitlines()
+    )

imap_processing/tests/ialirt/unit/test_process_ephemeris.py

@@ -0,0 +1,109 @@
+"""Test processEphemeris functions."""
+
+import numpy as np
+import pytest
+
+from imap_processing.ialirt import process_ephemeris
+from imap_processing.spice.time import str_to_et
+
+
+def test_calculate_doppler():
+    """
+    Test the calculate_doppler() function.
+    """
+    assert process_ephemeris.calculate_doppler(805794429.1837295) == 1
+
+
+@pytest.mark.external_kernel()
+def test_latitude_longitude_to_ecef(furnish_kernels):
+    """
+    Test the latitude_longitude_to_ecef() function.
+
+    Test data is from https://naif.jpl.nasa.gov/pub/naif/toolkit_docs/C/cspice/
+    georec_c.html.
+    """
+    longitude = 118.0  # degrees
+    latitude = 30.0  # degrees
+    altitude = 0.0  # kilometers
+
+    kernels = ["pck00011.tpc"]
+    with furnish_kernels(kernels):
+        rect_coords = process_ephemeris.latitude_longitude_to_ecef(
+            longitude, latitude, altitude
+        )
+
+    np.testing.assert_allclose(
+        rect_coords, [-2595.359123, 4881.160589, 3170.373523], atol=1e-6
+    )
+
+
+@pytest.mark.external_kernel()
+def test_calculate_azimuth_and_elevation(furnish_kernels):
+    """
+    Test the calculate_azimuth_and_elevation() function.
+    """
+    longitude = -71.41  # longitude in degrees
+    latitude = -33.94  # latitude in degrees
+    altitude = 0.157  # altitude in kilometers
+    # test single observation time
+    observation_time = 805794429.1837295  # "2025-07-14T19:46:00.000"
+
+    kernels = [
+        "pck00011.tpc",
+        "de440s.bsp",
+    ]
+    with furnish_kernels(kernels):
+        azimuth_result, elevation_result = (
+            process_ephemeris.calculate_azimuth_and_elevation(
+                longitude, latitude, altitude, observation_time
+            )
+        )
+    assert azimuth_result, elevation_result is not None
+
+    # test array of observation times
+    time_endpoints = ("2026 SEP 22 00:00:00", "2026 SEP 22 23:59:59")
+    time_interval = int(1e3)  # seconds between data points
+    observation_time = np.arange(
+        str_to_et(time_endpoints[0]), str_to_et(time_endpoints[1]), time_interval
+    )
+    with furnish_kernels(kernels):
+        azimuth_result, elevation_result = (
+            process_ephemeris.calculate_azimuth_and_elevation(
+                longitude, latitude, altitude, observation_time
+            )
+        )
+    assert len(azimuth_result) == len(observation_time)
+
+
+@pytest.mark.external_kernel()
+def test_build_output(furnish_kernels):
+    """
+    Test the build_output() function.
+    """
+    # Example usage for Longovilo-Hacienda, Chile
+    # https://www.findlatitudeandlongitude.com/l/LONGOVILO%2C+CHILE/
+    longitude = -71.41  # longitude in degrees
+    latitude = -33.94  # latitude in degrees
+    altitude = 0.157  # altitude in kilometers
+    time_endpoints = ("2026 SEP 22 00:00:00", "2026 SEP 22 23:59:59")
+    time_interval = int(1e3)  # seconds between data points
+
+    kernels = [
+        "naif0012.tls",
+        "pck00011.tpc",
+        "de440s.bsp",
+    ]
+    with furnish_kernels(kernels):
+        output_dict = process_ephemeris.build_output(
+            longitude, latitude, altitude, time_endpoints, time_interval
+        )
+
+    for key_name in ["azimuth", "elevation", "time", "doppler"]:
+        assert key_name in output_dict.keys()
+        assert len(output_dict[key_name]) == len(
+            np.arange(
+                str_to_et(time_endpoints[0]),
+                str_to_et(time_endpoints[1]),
+                time_interval,
+            )
+        )

imap_processing/tests/ialirt/unit/test_process_hit.py

@@ -20,7 +20,7 @@ def xtce_hit_path():
 
 @pytest.fixture(scope="session")
 def binary_packet_path():
-    """Returns the xtce auxiliary directory."""
+    """Returns the xtce directory."""
     return (
         imap_module_directory
         / "tests"
@@ -33,7 +33,7 @@ def binary_packet_path():
 
 @pytest.fixture(scope="session")
 def hit_test_data():
-    """Returns the xtce auxiliary directory."""
+    """Returns the test data directory."""
     data_path = (
         imap_module_directory
         / "tests"
@@ -141,7 +141,7 @@ def test_create_l1(xarray_data):
     assert l1["L4IBHG"] == 2
 
 
-def test_process_hit(xarray_data):
+def test_process_hit(xarray_data, caplog):
     """Tests process_hit."""
 
     # Tests that it functions normally
@@ -167,11 +167,14 @@ def test_process_hit(xarray_data):
         i for i in range(29) for _ in range(2)
     ] + [59, 59]
 
-    with pytest.raises(
-        ValueError, match="does not contain all values from 0 to 59 without duplicates"
-    ):
+    with caplog.at_level("WARNING"):
         process_hit(subset)
 
+    assert any(
+        "does not contain all values from 0 to 59 without duplicates" in message
+        for message in caplog.text.splitlines()
+    )
+
 
 def test_decom_packets(xarray_data, hit_test_data):
     """This function checks that all instrument parameters are accounted for."""

imap_processing/tests/idex/conftest.py

@@ -7,7 +7,7 @@ from imap_processing import imap_module_directory
 from imap_processing.idex.idex_l1a import PacketParser
 
 
-@pytest.fixture()
+@pytest.fixture(scope="module")
 def decom_test_data() -> xr.Dataset:
     """Return a ``xarray`` dataset containing test data.
 
@@ -17,6 +17,6 @@ def decom_test_data() -> xr.Dataset:
         A ``xarray`` dataset containing the test data
     """
     test_file = Path(
-        f"{imap_module_directory}/tests/idex/imap_idex_l0_raw_20230725_v001.pkts"
+        f"{imap_module_directory}/tests/idex/imap_idex_l0_raw_20231214_v001.pkts"
     )
     return PacketParser(test_file, "001").data