imap-processing 0.6.0__py3-none-any.whl

imap_processing/tests/cdf/test_cdf_attribute_manager.py

@@ -0,0 +1,353 @@
+from pathlib import Path
+
+import pytest
+
+from imap_processing.cdf.cdf_attribute_manager import CdfAttributeManager
+
+
+def test_default_attr_schema():
+    """
+    Test function that covers:
+        _load_default_global_attr_schema
+        _load_default_variable_attr_schema
+    """
+
+    # Initialize CdfAttributeManager object which loads in default schema
+    cdf_manager = CdfAttributeManager(Path(__file__).parent.parent / "cdf")
+
+    # Default global tests
+    assert cdf_manager.global_attribute_schema["DOI"]["required"] is False
+    assert cdf_manager.global_attribute_schema["Data_level"]["required"] is True
+
+    # Default variable tests
+    assert (
+        cdf_manager.variable_attribute_schema["attribute_key"]["ABSOLUTE_ERROR"][
+            "required"
+        ]
+        is False
+    )
+    assert (
+        cdf_manager.variable_attribute_schema["attribute_key"]["RESOLUTION"]["required"]
+        is True
+    )
+
+
+def test_load_global_attribute():
+    """
+    Test function that covers:
+        load_global_attributes
+    """
+
+    # Initialize CdfAttributeManager object which loads in default info
+    cdf_manager = CdfAttributeManager(Path(__file__).parent.parent / "cdf")
+    cdf_manager.load_global_attributes("imap_default_global_cdf_attrs.yaml")
+
+    # Testing information has been loaded in
+    assert cdf_manager._global_attributes["Project"] == "STP>Solar-Terrestrial Physics"
+    assert (
+        cdf_manager._global_attributes["Source_name"]
+        == "IMAP>Interstellar Mapping and Acceleration Probe"
+    )
+    assert (
+        cdf_manager._global_attributes["Discipline"]
+        == "Solar Physics>Heliospheric Physics"
+    )
+    assert (
+        cdf_manager._global_attributes["Mission_group"]
+        == "IMAP>Interstellar Mapping and Acceleration Probe"
+    )
+    assert cdf_manager._global_attributes["PI_name"] == "Dr. David J. McComas"
+    assert (
+        cdf_manager._global_attributes["PI_affiliation"]
+        == "Princeton Plasma Physics Laboratory, 100 Stellarator Road, "
+        "Princeton, NJ 08540"
+    )
+    assert (
+        cdf_manager._global_attributes["File_naming_convention"]
+        == "source_descriptor_datatype_yyyyMMdd_vNNN"
+    )
+    # The following test should fail because "DOI" is not an attribute in
+    #   imap_default_global_cdf_attrs.yaml
+    with pytest.raises(KeyError):
+        assert cdf_manager._global_attributes["DOI"] == "test"
+
+    # Load in different data
+    cdf_manager.source_dir = Path(__file__).parent.parent / "cdf" / "test_data"
+    cdf_manager.load_global_attributes("imap_default_global_test_cdf_attrs.yaml")
+
+    # Testing attributes carried over
+    assert (
+        cdf_manager._global_attributes["File_naming_convention"]
+        == "source_descriptor_datatype_yyyyMMdd_vNNN"
+    )
+    assert (
+        cdf_manager._global_attributes["Discipline"]
+        == "Solar Physics>Heliospheric Physics"
+    )
+
+    # Testing attributes newly loaded
+    assert cdf_manager._global_attributes["Project"] == "STP>Solar-Terrestrial Physics"
+    assert (
+        cdf_manager._global_attributes["Source_name"]
+        == "IMAP>Interstellar Mapping and Acceleration Probe"
+    )
+    assert cdf_manager._global_attributes["Mission_group"] == "Dysfunctional Cats"
+    assert cdf_manager._global_attributes["PI_name"] == "Ana Manica"
+    assert cdf_manager._global_attributes["PI_affiliation"] == "LASP, CU"
+    assert cdf_manager._global_attributes["Data_version"] == 2
+    assert cdf_manager._global_attributes["DOI"] == "test"
+
+    # Testing that everything loaded into the global attrs is present in
+    #   the global attrs schema
+    for attr in cdf_manager._global_attributes.keys():
+        assert attr in cdf_manager.global_attribute_schema.keys()
+
+
+def test_get_global_attributes():
+    """
+    Test function that covers:
+        get_global_attributes
+    """
+
+    # Initialize CdfAttributeManager object which loads in default info
+    cdf_manager = CdfAttributeManager(Path(__file__).parent.parent / "cdf")
+
+    # Change filepath to load test global attributes
+    cdf_manager.source_dir = Path(__file__).parent.parent / "cdf" / "test_data"
+    cdf_manager.load_global_attributes("imap_default_global_test_cdf_attrs.yaml")
+    cdf_manager.load_global_attributes("imap_test_global.yaml")
+
+    # Loading in instrument specific attributes
+    test_get_global_attrs = cdf_manager.get_global_attributes("imap_test_T1_test")
+
+    # Testing information previously loaded into global attributes (first if statement)
+    assert test_get_global_attrs["Project"] == "STP>Solar-Terrestrial Physics"
+    assert (
+        test_get_global_attrs["Source_name"]
+        == "IMAP>Interstellar Mapping and Acceleration Probe"
+    )
+    assert test_get_global_attrs["Mission_group"] == "Dysfunctional Cats"
+    # Testing instrument specific global attributes (first elif statement)
+    assert test_get_global_attrs["Descriptor"] == "TEST>Testinstrument"
+    assert test_get_global_attrs["Data_type"] == "T1_test-one>Test-1 test one"
+    assert test_get_global_attrs["Logical_source"] == "imap_test_T1_test"
+    assert (
+        test_get_global_attrs["Logical_source_description"]
+        == "IMAP Mission TEST one document Level-T1."
+    )
+    # Not given, and not required information
+    assert test_get_global_attrs["Data_level"] is None
+    with pytest.raises(KeyError):
+        assert test_get_global_attrs["bad_name"] == "false info"
+
+    # Testing second elif statement
+    test_error_elif = cdf_manager.get_global_attributes("imap_test_T3_test")
+    with pytest.raises(KeyError):
+        assert test_error_elif["Data_type"] == "Does Not Exist"
+
+    # Load in more data using get_global_attributes
+    test_get_global_attrs_2 = cdf_manager.get_global_attributes("imap_test_T2_test")
+    # Testing information previously loaded into global attributes (first if statement)
+    assert test_get_global_attrs_2["Project"] == "STP>Solar-Terrestrial Physics"
+    # Testing first elif statement
+    assert test_get_global_attrs_2["Descriptor"] == "TEST>Testinstrument"
+    # "Data_type" not required according to default schema
+    assert test_get_global_attrs_2["Data_type"] == "T2_test-two>Test-2 test two"
+
+    # Testing that required schema keys are in get_global_attributes
+    for attr_name in cdf_manager.global_attribute_schema.keys():
+        required_schema = cdf_manager.global_attribute_schema[attr_name]["required"]
+        if required_schema is True:
+            assert attr_name in test_get_global_attrs.keys()
+
+
+def test_instrument_id_format():
+    # Initialize CdfAttributeManager object which loads in default info
+    cdf_manager = CdfAttributeManager(Path(__file__).parent.parent / "cdf")
+
+    # Change filepath to load test global attributes
+    cdf_manager.source_dir = Path(__file__).parent.parent / "cdf" / "test_data"
+    cdf_manager.load_global_attributes("imap_default_global_test_cdf_attrs.yaml")
+    cdf_manager.load_global_attributes("imap_test_global.yaml")
+
+    # Loading in instrument specific attributes
+    test_get_global_attrs = cdf_manager.get_global_attributes("imap_test_T1_test")
+
+    # Testing how instrument_id operates
+    assert test_get_global_attrs["Project"] == cdf_manager._global_attributes["Project"]
+    assert (
+        test_get_global_attrs["Source_name"]
+        == cdf_manager._global_attributes["Source_name"]
+    )
+    assert (
+        test_get_global_attrs["Data_type"]
+        == cdf_manager._global_attributes["imap_test_T1_test"]["Data_type"]
+    )
+    assert (
+        cdf_manager._global_attributes["imap_test_T1_test"]["Logical_source"]
+        == "imap_test_T1_test"
+    )
+    with pytest.raises(KeyError):
+        assert cdf_manager._global_attributes["imap_test_T1_test"]["Project"]
+
+
+def test_add_global_attribute():
+    # Initialize CdfAttributeManager object which loads in default info
+    cdf_manager = CdfAttributeManager(Path(__file__).parent.parent / "cdf")
+
+    # Change filepath to load test global attributes
+    cdf_manager.source_dir = Path(__file__).parent.parent / "cdf" / "test_data"
+    cdf_manager.load_global_attributes("imap_test_global.yaml")
+
+    # Changing a dynamic global variable
+    cdf_manager.add_global_attribute("Project", "Test Project")
+    test_get_global_attrs = cdf_manager.get_global_attributes("imap_test_T1_test")
+    assert cdf_manager._global_attributes["Project"] == "Test Project"
+    assert test_get_global_attrs["Project"] == "Test Project"
+
+    # Testing adding required global attribute
+    cdf_manager._global_attributes.__delitem__("Source_name")
+    # Reloading get_global_attributes to pick up deleted Source_name
+    test_get_global_attrs = cdf_manager.get_global_attributes("imap_test_T1_test")
+    with pytest.raises(KeyError):
+        assert cdf_manager._global_attributes["Source_name"]
+    assert test_get_global_attrs["Source_name"] is None
+
+    # Adding deleted global attribute
+    cdf_manager.add_global_attribute("Source_name", "anas_source")
+    assert cdf_manager._global_attributes["Source_name"] == "anas_source"
+    # Reloading get_global_attributes to pick up added Source_name
+    test_get_global_attrs = cdf_manager.get_global_attributes("imap_test_T1_test")
+    assert test_get_global_attrs["Source_name"] == "anas_source"
+
+    # Testing instrument specific attribute
+    cdf_manager._global_attributes["imap_test_T1_test"].__delitem__("Logical_source")
+    # Reloading get_global_attributes to pick up deleted Source_name
+    test_get_global_attrs = cdf_manager.get_global_attributes("imap_test_T1_test")
+    with pytest.raises(KeyError):
+        assert cdf_manager._global_attributes["imap_test_T1_test"]["Logical_source"]
+    assert test_get_global_attrs["Logical_source"] is None
+
+
+def test_variable_attribute():
+    """
+    Test function that covers:
+        load_variable_attributes
+        get_variable_attributes
+    """
+
+    # Creating CdfAttributeManager object, loading in default data
+    cdf_manager = CdfAttributeManager(Path(__file__).parent.parent / "cdf")
+    cdf_manager.source_dir = Path(__file__).parent.parent / "cdf" / "test_data"
+    cdf_manager.load_global_attributes("imap_default_global_test_cdf_attrs.yaml")
+    # Loading in test data
+    cdf_manager.load_variable_attributes("imap_test_variable.yaml")
+
+    # All variables required to have:
+    expected_attributes = [
+        "CATDESC",
+        "DEPEND_0",
+        "DISPLAY_TYPE",
+        "FIELDNAM",
+        "FILLVAL",
+        "FORMAT",
+        "LABLAXIS",
+        "UNITS",
+        "VALIDMIN",
+        "VALIDMAX",
+        "VAR_TYPE",
+    ]
+
+    # Assuring all required attributes are loaded in
+    for attr_name in cdf_manager.variable_attribute_schema["attribute_key"]:
+        attribute = cdf_manager.variable_attribute_schema["attribute_key"][attr_name]
+        if attribute["required"] is True:
+            for exp_attr in expected_attributes:
+                assert (
+                    exp_attr in cdf_manager.variable_attribute_schema["attribute_key"]
+                )
+
+    # Testing specific attributes
+    assert (
+        cdf_manager._variable_attributes["default_attrs"]["DEPEND_0"]
+        == cdf_manager._variable_attributes["default_attrs"]["DEPEND_0"]
+    )
+    assert cdf_manager._variable_attributes["default_attrs"]["FILLVAL"] == -10
+    assert cdf_manager._variable_attributes["test_field_1"]["DEPEND_0"] == "test_depend"
+    assert (
+        cdf_manager._variable_attributes["default_attrs"]["VAR_TYPE"] == "test_var_type"
+    )
+    with pytest.raises(KeyError):
+        assert cdf_manager._variable_attributes["default_attrs"]["CATDESC"] == "test"
+
+
+def test_get_variable_attributes():
+    # Creating CdfAttributeManager object, loading in default data
+    cdf_manager = CdfAttributeManager(Path(__file__).parent.parent / "cdf")
+
+    # Change filepath to load test global attributes
+    cdf_manager.source_dir = Path(__file__).parent.parent / "cdf" / "test_data"
+    cdf_manager.load_global_attributes("imap_default_global_test_cdf_attrs.yaml")
+    cdf_manager.load_variable_attributes("imap_test_variable.yaml")
+
+    # Loading in instrument specific attributes
+    imap_test_variable = cdf_manager.get_variable_attributes("test_field_1")
+
+    # Make sure all expected attributes are present
+    for variable_attrs in cdf_manager.variable_attribute_schema["attribute_key"]:
+        required_var_attributes = cdf_manager.variable_attribute_schema[
+            "attribute_key"
+        ][variable_attrs]["required"]
+        if required_var_attributes is True:
+            assert variable_attrs in imap_test_variable.keys()
+
+    # Calling default attributes
+    assert imap_test_variable["DEPEND_0"] == "test_depend"
+    assert imap_test_variable["DISPLAY_TYPE"] == "test_display_type"
+    assert imap_test_variable["FILLVAL"] == -10
+
+    # Calling required attributes
+    assert imap_test_variable["CATDESC"] == "test time"
+    assert imap_test_variable["TIME_BASE"] == 10
+    assert imap_test_variable["DEPEND_1"] == "test_depend_1"
+    assert imap_test_variable["DEPEND_2"] == "test_depend_2"
+
+    # Calling to non required attributes
+    assert imap_test_variable["LEAP_SECONDS_INCLUDED"] == "test_not_required"
+
+    # Calling attribute name that does not exist
+    with pytest.raises(KeyError):
+        assert imap_test_variable["DOES_NOT_EXIST"]
+
+    # Testing for attribute not in schema
+    with pytest.raises(KeyError):
+        assert imap_test_variable["NOT_IN_SCHEMA"]
+
+    # Load in different data, test again
+    imap_test_variable_2 = cdf_manager.get_variable_attributes("test_field_2")
+    # Calling default attributes
+    assert imap_test_variable_2["DEPEND_0"] == "test_depend"
+    assert imap_test_variable_2["DISPLAY_TYPE"] == "test_display_type"
+    assert imap_test_variable_2["FILLVAL"] == -10
+
+    # Calling required attributes
+    assert imap_test_variable_2["CATDESC"] == "test time 2"
+    assert imap_test_variable_2["TIME_BASE"] == 11
+
+    # Loading in different data to test logger errors, empty strings,
+    # and DEPEND_i with i >= 1 condition
+    imap_test_variable_3 = cdf_manager.get_variable_attributes("test_field_3")
+
+    assert imap_test_variable_3["DEPEND_1"] == "depend_1_test_3"
+    with pytest.raises(KeyError):
+        assert imap_test_variable_3["DEPEND_0"]
+    assert imap_test_variable_3["CATDESC"] == ""
+    assert imap_test_variable_3["REPRESENTATION_2"] == "representation_2"
+    assert imap_test_variable_3["LABL_PTR_1"] == "labl_ptr_1"
+
+    # check_schema = False
+    imap_test_variable_1_false = cdf_manager.get_variable_attributes(
+        "test_field_1", False
+    )
+    assert imap_test_variable_1_false["NOT_IN_SCHEMA"] == "not_in_schema"
+    assert imap_test_variable_1_false["VALIDMIN"] == 0

imap_processing/tests/cdf/test_data/imap_default_global_test_cdf_attrs.yaml

@@ -0,0 +1,7 @@
+Project: STP>Solar-Terrestrial Physics
+Source_name: IMAP>Interstellar Mapping and Acceleration Probe
+Mission_group: Dysfunctional Cats
+PI_name: Ana Manica
+PI_affiliation: LASP, CU
+Data_version: 002
+DOI: test

imap_processing/tests/cdf/test_data/imap_instrument1_global_cdf_attrs.yaml

@@ -0,0 +1,14 @@
+Descriptor: &desc
+  TEST>Testinstrument
+TEXT: &txt >
+  This file is nothing more than a test yamal. This description? This is a test description.
+  This Testinstrument will contribute to our understanding of the cdf_attribute_manager.py file.
+  This test file is led by me, Ana Manica. Lasp undergraduate employee.
+  I don't have a website. Sorry.
+
+imap_test_T1_test:
+    Descriptor: *desc
+    TEXT: *txt
+    Data_type: T1_test-one>Test-1 test one
+    Logical_source: imap_test_T1_test
+    Logical_source_description: IMAP Mission TEST one document Level-T1.

imap_processing/tests/cdf/test_data/imap_instrument1_level1_variable_attrs.yaml

@@ -0,0 +1,23 @@
+default_attrs: &default
+  # Assumed values for all variable attrs unless overwritten
+  DEPEND_0: test_depend
+  DISPLAY_TYPE: test_display_type
+  FILLVAL: -10
+  FORMAT: I1
+  VALIDMIN: 0
+  VALIDMAX: 10
+  VAR_TYPE: test_var_type
+
+test_field_1:
+  <<: *default
+  CATDESC: test time
+  FIELDNAM: test_field_1
+  LABLAXIS: test_labaxis
+  UNITS: test_units
+  VAR_TYPE: test_variable_type
+  SCALETYP: test_scaletyp
+  MONOTON: test_monoton
+  TIME_BASE: 10
+  TIME_SCALE: test_time_scale
+  REFERENCE_POSITION: test_reference_position
+  NOT_REQUIRED: test_not_required

imap_processing/tests/cdf/test_data/imap_instrument2_global_cdf_attrs.yaml

@@ -0,0 +1,23 @@
+instrument_base: &instrument_base
+  Descriptor: SWE>Solar Wind Electron
+  TEXT: >
+    The IMAP Solar Wind Electron (SWE) instrument measures the solar wind electrons
+    in the heliosphere. SWE will contribute to our understanding of the acceleration
+    and transport of charged particles in the heliosphere.
+    SWE design and assembly is led by the Princeton Plasma Physics Laboratory. See
+    https://imap.princeton.edu/instruments/swe for more details.
+  Instrument_type: "Particles (space)"
+
+imap_swe_l1a_sci:
+  <<: *instrument_base
+  Data_level: 1A
+  Data_type: L1A_SCI>Level-1A Science data
+  Logical_source: imap_swe_l1a_sci
+  Logical_source_description: SWE Instrument Level-1A Science Data
+
+imap_swe_l1b_sci:
+  <<: *instrument_base
+  Data_level: 1A
+  Data_type: L1B_SCI>Level-1B Science data
+  Logical_source: imap_swe_l1b_sci
+  Logical_source_description: SWE Instrument Level-1B Science Data

imap_processing/tests/cdf/test_data/imap_instrument2_level2_variable_attrs.yaml

@@ -0,0 +1,30 @@
+default_attrs: &default
+  # Assumed values for all variable attrs unless overwritten
+  DEPEND_0: epoch
+  DISPLAY_TYPE: time_series
+  FILLVAL: -9223372036854775808
+  FORMAT: I19
+  VALIDMIN: 0
+  VALIDMAX: 9223372036854775807
+  VAR_TYPE: data
+
+epoch:
+  <<: *default
+  CATDESC: Time, number of nanoseconds since J2000 with leap seconds included
+  FIELDNAM: epoch
+  LABLAXIS: epoch
+  UNITS: ns
+  VAR_TYPE: support_data
+  SCALETYP: linear
+  MONOTON: INCREASE
+  TIME_BASE: J2000
+  TIME_SCALE: Terrestrial Time
+  REFERENCE_POSITION: Rotating Earth Geoid
+
+x_front:
+  <<: *default
+  CATDESC: x front position
+  FIELDNAM: Event x-position on front foil
+  LABLAXIS: x front position
+  # TODO: come back to format
+  UNITS: mm

imap_processing/tests/cdf/test_data/imap_test_global.yaml

@@ -0,0 +1,26 @@
+Descriptor: &desc
+  TEST>Testinstrument
+TEXT: &txt >
+  This file is nothing more than a test yamal. This description? This is a test description.
+  This Testinstrument will contribute to our understanding of the cdf_attribute_manager.py file.
+  This test file is led by me, Ana Manica. Lasp undergraduate employee.
+  I don't have a website. Sorry.
+
+imap_test_T1_test:
+    Descriptor: *desc
+    TEXT: *txt
+    Data_type: T1_test-one>Test-1 test one
+    Logical_source: imap_test_T1_test
+    Logical_source_description: IMAP Mission TEST one document Level-T1.
+
+imap_test_T2_test:
+    Descriptor: *desc
+    TEXT: *txt
+    Data_type: T2_test-two>Test-2 test two
+    Logical_source: imap_test_T2_test
+    Logical_source_description: IMAP Mission TEST two document Level-T2.
+    LINK_TEXT: Test two additional info
+
+imap_test_T3_test:
+    Descriptor: *desc
+    TEXT: *txt

imap_processing/tests/cdf/test_data/imap_test_variable.yaml

@@ -0,0 +1,41 @@
+default_attrs: &default
+  # Assumed values for all variable attrs unless overwritten
+  DEPEND_0: test_depend
+  DISPLAY_TYPE: test_display_type
+  FILLVAL: -10
+  FORMAT: I1
+  VALIDMIN: 0
+  VALIDMAX: 10
+  VAR_TYPE: test_var_type
+
+test_field_1:
+  <<: *default
+  CATDESC: test time
+  FIELDNAM: test_field_1
+  LABLAXIS: test_labaxis
+  UNITS: test_units
+  VAR_TYPE: test_variable_type
+  TIME_BASE: 10
+  TIME_SCALE: test_time_scale
+  REFERENCE_POSITION: test_reference_position
+  LEAP_SECONDS_INCLUDED: test_not_required
+  DEPEND_1: test_depend_1
+  DEPEND_2: test_depend_2
+  NOT_IN_SCHEMA: not_in_schema
+
+test_field_2:
+  <<: *default
+  CATDESC: test time 2
+  FIELDNAM: test_field_2
+  LABLAXIS: test_labaxis_2
+  UNITS: test_units_2
+  VAR_TYPE: test_variable_type_2
+  TIME_BASE: 11
+  TIME_SCALE: test_time_scale_2
+  REFERENCE_POSITION: test_reference_position_2
+
+test_field_3:
+  DEPEND_1: depend_1_test_3
+  DEPEND_1: depend_1_test_3
+  LABL_PTR_1: labl_ptr_1
+  REPRESENTATION_2: representation_2

imap_processing/tests/cdf/test_imap_cdf_manager.py

@@ -0,0 +1,62 @@
+from pathlib import Path
+from unittest import mock
+
+# from imap_processing.cdf.cdf_attribute_manager import CdfAttributeManager
+from imap_processing.cdf.imap_cdf_manager import ImapCdfAttributes
+
+
+def test_add_instrument_global_attrs():
+    # Create an ImapCdfAttributes object, set to correct file path
+    imap_cdf_manager = ImapCdfAttributes()
+    imap_cdf_manager.source_dir = Path(__file__).parent.parent / "cdf" / "test_data"
+    imap_cdf_manager.add_instrument_global_attrs("instrument1")
+
+    # Testing data loaded in
+    imap_instrument = imap_cdf_manager.get_global_attributes("imap_test_T1_test")
+    assert imap_instrument["Data_type"] == "T1_test-one>Test-1 test one"
+    assert imap_instrument["Project"] == "STP>Solar Terrestrial Probes"
+
+    # Testing reloading data
+    imap_cdf_manager.add_instrument_global_attrs("instrument2")
+
+    # Testing data carried over, and overwritten
+    instrument2_instrument = imap_cdf_manager.get_global_attributes("imap_swe_l1a_sci")
+    assert instrument2_instrument["Data_type"] == "L1A_SCI>Level-1A Science data"
+    assert instrument2_instrument["Project"] == "STP>Solar Terrestrial Probes"
+
+
+@mock.patch(
+    "imap_processing.cdf.cdf_attribute_manager.CdfAttributeManager.load_variable_attributes"
+)
+def testing_source_dir(mock_load_variable_attributes):
+    # Create an ImapCdfAttributes object
+    imap_cdf_manager = ImapCdfAttributes(Path(__file__).parent.parent / "cdf")
+    assert str(imap_cdf_manager.source_dir) == str(Path(__file__).parent.parent / "cdf")
+
+
+def test_add_instrument_variable_attrs():
+    # Create an ImapCdfAttributes object
+    imap_cdf_manager = ImapCdfAttributes()
+    imap_cdf_manager.source_dir = Path(__file__).parent.parent / "cdf" / "test_data"
+    imap_cdf_manager.add_instrument_variable_attrs("instrument1", "level1")
+
+    # Testing the actual function
+    imap_instrument = imap_cdf_manager.get_variable_attributes("test_field_1")
+    assert imap_instrument["DEPEND_0"] == "test_depend"
+    assert imap_instrument["CATDESC"] == "test time"
+    assert imap_instrument["FIELDNAM"] == "test_field_1"
+    assert imap_instrument["UNITS"] == "test_units"
+
+    # Testing reloading data
+    imap_cdf_manager.add_instrument_variable_attrs("instrument2", "level2")
+
+    # Testing again
+    instrument2_instrument = imap_cdf_manager.get_variable_attributes("epoch")
+    assert instrument2_instrument["DEPEND_0"] == "epoch"
+    assert instrument2_instrument["DISPLAY_TYPE"] == "time_series"
+    assert instrument2_instrument["VALIDMAX"] == 9223372036854775807
+    assert (
+        instrument2_instrument["CATDESC"]
+        == "Time, number of nanoseconds since J2000 with leap seconds included"
+    )
+    assert instrument2_instrument["UNITS"] == "ns"