Skip to content
Snippets Groups Projects
Select Git revision
  • 8a2ce12161a1de99a4bcb18473c5a9ed04104e8f
  • master default protected
  • ia2
  • adql2.1-ia2
  • private_rows
5 results

JDBCTranslator.java

Blame
  • test_mdis_drivers.py 9.32 KiB
    import pytest
    import os
    import numpy as np
    import spiceypy as spice
    from importlib import reload
    import json
    import unittest
    from unittest.mock import patch
    
    from conftest import get_image_label, get_image_kernels, convert_kernels, compare_dicts
    import ale
    from ale.drivers.mes_drivers import MessengerMdisPds3NaifSpiceDriver
    from ale.drivers.mes_drivers import MessengerMdisIsisLabelNaifSpiceDriver
    
    @pytest.fixture(scope='module')
    def test_kernels():
        kernels = get_image_kernels('EN1072174528M')
        updated_kernels, binary_kernels = convert_kernels(kernels)
        yield updated_kernels
        for kern in binary_kernels:
            os.remove(kern)
    
    @pytest.fixture()
    def usgscsm_compare_dict():
        return {
        'radii': {
            'semimajor': 2439.4,
            'semiminor': 2439.4,
            'unit': 'km'},
        'sensor_position': {
            'positions': np.array([[-629657.4559945846, -1583350.7374122413, 1784408.773440049]]),
            'velocities': np.array([[1732.8734290653545, 2504.0213215928925, 2412.578186708735]]),
            'unit': 'm'},
        'sun_position': {
            'positions': np.array([[-4.68946673e+10, -5.36158427e+08,  2.71167863e+07]]),
            'velocities': np.array([[-4629.73346128, 256.72086237, 10.63960444]]),
            'unit': 'm'},
        'sensor_orientation': {
            'quaternions': np.array([[ 0.93418372,  0.00144773, -0.00449382, -0.35676112]])},
        'detector_sample_summing': 2,
        'detector_line_summing': 2,
        'focal_length_model': {
            'focal_length': 549.5535053027719},
        'detector_center': {
            'line': 512,
            'sample': 512},
        'starting_detector_line': 1,
        'starting_detector_sample': 9,
        'focal2pixel_lines': [0.0, 0.0, 71.42857143],
        'focal2pixel_samples': [0.0, 71.42857143, 0.0],
        'optical_distortion': {
            'transverse': {
                'x': [0.0, 1.001854269623802, 0.0, 0.0, -0.0005094440474941111, 0.0, 1.004010471468856e-05, 0.0, 1.004010471468856e-05, 0.0],
                'y': [0.0, 0.0, 1.0, 0.0009060010594996751, 0.0, 0.0003574842626620758, 0.0, 1.004010471468856e-05, 0.0, 1.004010471468856e-05]}},
        'image_lines': 512,
        'image_samples': 512,
        'name_platform': 'MESSENGER',
        'name_sensor': 'MERCURY DUAL IMAGING SYSTEM NARROW ANGLE CAMERA',
        'reference_height': {
            'maxheight': 1000,
            'minheight': -1000,
            'unit': 'm'},
        'name_model': 'USGS_ASTRO_FRAME_SENSOR_MODEL',
        'center_ephemeris_time': 483122606.85252464}
    
    @pytest.mark.parametrize("label_type", ["pds3", "isis3"])
    def test_load(test_kernels, usgscsm_compare_dict, label_type):
        label_file = get_image_label('EN1072174528M', label_type)
    
        usgscsm_isd_str = ale.loads(label_file, props={'kernels': test_kernels}, formatter='usgscsm')
        usgscsm_isd_obj = json.loads(usgscsm_isd_str)
    
        assert compare_dicts(usgscsm_isd_obj, usgscsm_compare_dict) == []
    
    # ========= Test Pds3 Label and NAIF Spice driver =========
    class test_pds3_naif(unittest.TestCase):
    
        def setUp(self):
            label = get_image_label("EN1072174528M", "pds3")
            self.driver = MessengerMdisPds3NaifSpiceDriver(label)
    
        def test_short_mission_name(self):
            assert self.driver.short_mission_name=='mes'
    
        def test_spacecraft_name(self):
            assert self.driver.spacecraft_name == 'MESSENGER'
    
        def test_fikid(self):
            with patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.spacecraft_name == 'MESSENGER'
    
        def test_instrument_id(self):
            assert self.driver.instrument_id == 'MSGR_MDIS_NAC'
    
        def test_sampling_factor(self):
            assert self.driver.sampling_factor == 2
    
        def test_focal_length(self):
            with patch('ale.drivers.mes_drivers.spice.gdpool', return_value=np.array([pow(4.07, -x) for x in np.arange(6)])) as gdpool, \
                 patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.focal_length == pytest.approx(6.0)
                gdpool.assert_called_with('INS-12345_FL_TEMP_COEFFS', 0, 6)
    
        def test_detector_start_sample(self):
            with patch('ale.drivers.mes_drivers.spice.gdpool', return_value=np.array([10.0])) as gdpool, \
                 patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.detector_start_sample == 10.0
                gdpool.assert_called_with('INS-12345_FPUBIN_START_SAMPLE', 0, 1)
    
        def test_detector_start_line(self):
            with patch('ale.drivers.mes_drivers.spice.gdpool', return_value=np.array([10.0])) as gdpool, \
                 patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.detector_start_line == 10.0
                gdpool.assert_called_with('INS-12345_FPUBIN_START_LINE', 0, 1)
    
        def test_detector_center_sample(self):
            assert self.driver.detector_center_sample == 512
    
        def test_detector_center_line(self):
            assert self.driver.detector_center_line == 512
    
        def test_sensor_model_version(self):
            assert self.driver.sensor_model_version == 2
    
        def test_usgscsm_distortion_model(self):
            with patch('ale.drivers.mes_drivers.spice.gdpool', side_effect=[np.array([1, 2, 3, 4, 5]), np.array([-1, -2, -3, -4, -5])]) as gdpool, \
                 patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.usgscsm_distortion_model == {"transverse" : {
                                                                    "x" : [1, 2, 3, 4, 5],
                                                                    "y" : [-1, -2, -3, -4, -5]}}
    
    
    
        def test_pixel_size(self):
            with patch('ale.drivers.mes_drivers.spice.gdpool', return_value=np.array([0.1])) as gdpool, \
                 patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.pixel_size == 0.1
                gdpool.assert_called_with('INS-12345_PIXEL_PITCH', 0, 1)
    
    # ========= Test ISIS3 Label and NAIF Spice driver =========
    class test_isis3_naif(unittest.TestCase):
    
        def setUp(self):
            label = get_image_label("EN1072174528M", "isis3")
            self.driver = MessengerMdisIsisLabelNaifSpiceDriver(label)
    
        def test_short_mission_name(self):
            assert self.driver.short_mission_name=='mes'
    
        def test_platform_name(self):
            assert self.driver.platform_name == 'MESSENGER'
    
        def test_fikid(self):
            with patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.spacecraft_name == 'MESSENGER'
    
        def test_instrument_id(self):
            assert self.driver.instrument_id == 'MSGR_MDIS_NAC'
    
        def test_sampling_factor(self):
            assert self.driver.sampling_factor == 2
    
        def test_focal_length(self):
            with patch('ale.drivers.mes_drivers.spice.gdpool', return_value=np.array([pow(4.07, -x) for x in np.arange(6)])) as gdpool, \
                 patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.focal_length == pytest.approx(6.0)
                gdpool.assert_called_with('INS-12345_FL_TEMP_COEFFS', 0, 6)
    
        def test_detector_start_sample(self):
            with patch('ale.drivers.mes_drivers.spice.gdpool', return_value=np.array([10.0])) as gdpool, \
                 patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.detector_start_sample == 10.0
                gdpool.assert_called_with('INS-12345_FPUBIN_START_SAMPLE', 0, 1)
    
        def test_detector_start_line(self):
            with patch('ale.drivers.mes_drivers.spice.gdpool', return_value=np.array([10.0])) as gdpool, \
                 patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.detector_start_line == 10.0
                gdpool.assert_called_with('INS-12345_FPUBIN_START_LINE', 0, 1)
    
        def test_detector_center_sample(self):
            with patch('ale.drivers.mes_drivers.spice.gdpool', return_value=np.array([512.5, 512.5, 1])) as gdpool, \
                 patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.detector_center_sample == 512
                gdpool.assert_called_with('INS-12345_CCD_CENTER', 0, 3)
    
        def test_detector_center_line(self):
            with patch('ale.drivers.mes_drivers.spice.gdpool', return_value=np.array([512.5, 512.5, 1])) as gdpool, \
                 patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.detector_center_line == 512
                gdpool.assert_called_with('INS-12345_CCD_CENTER', 0, 3)
    
        def test_sensor_model_version(self):
            assert self.driver.sensor_model_version == 2
    
        def test_usgscsm_distortion_model(self):
            with patch('ale.drivers.mes_drivers.spice.gdpool', side_effect=[np.array([1, 2, 3, 4, 5]), np.array([-1, -2, -3, -4, -5])]) as gdpool, \
                 patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.usgscsm_distortion_model == {"transverse" : {
                                                                    "x" : [1, 2, 3, 4, 5],
                                                                    "y" : [-1, -2, -3, -4, -5]}}
    
    
    
        def test_pixel_size(self):
            with patch('ale.drivers.mes_drivers.spice.gdpool', return_value=np.array([0.1])) as gdpool, \
                 patch('ale.base.data_naif.spice.bods2c', return_value=-12345) as bods2c:
                assert self.driver.pixel_size == 0.1
                gdpool.assert_called_with('INS-12345_PIXEL_PITCH', 0, 1)