Test that azimuth angles are added

foua-pps · Mar 8, 2024 · 2532e58 · 2532e58
1 parent f40368a
commit 2532e58
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/level1c4pps/seviri2pps_lib.py b/level1c4pps/seviri2pps_lib.py
@@ -611,10 +611,12 @@ def process_one_scan(tslot_files, out_path, rotate=True, engine='h5netcdf',
     update_coords(scn_)
 
     # Add ancillary datasets to the scene
-    add_ancillary_datasets(scn_, lons=lons, lats=lats, sunz=sunz, satz=satz,
+    add_ancillary_datasets(scn_,
+                           lons=lons, lats=lats,
+                           sunz=sunz, satz=satz,
+                           azidiff=azidiff,
                            suna=suna, sata=sata,
-                           save_azimuth_angles=save_azimuth_angles,
-                           azidiff=azidiff)
+                           save_azimuth_angles=save_azimuth_angles)
     add_proj_satpos(scn_)
 
     # Set attributes. This changes SEVIRI band names to PPS band names.

diff --git a/level1c4pps/tests/test_seviri2pps.py b/level1c4pps/tests/test_seviri2pps.py
@@ -312,10 +312,12 @@ def test_add_ancillary_datasets(self):
                                      'orbital_parameters': 'orb_params',
                                      'georef_offset_corrected': True})
         scene = {'IR_108': ir_108}
-        seviri2pps.add_ancillary_datasets(scene, lons=lons, lats=lats,
+        seviri2pps.add_ancillary_datasets(scene,
+                                          lons=lons, lats=lats,
                                           sunz=sunz, satz=satz,
+                                          azidiff=azidiff,
                                           suna=suna, sata=sata,
-                                          azidiff=azidiff)
+                                          save_azimuth_angles=True)
 
         # Test lon/lat
         np.testing.assert_array_equal(scene['lon'].data, lons)
@@ -334,6 +336,12 @@ def test_add_ancillary_datasets(self):
         np.testing.assert_array_equal(scene['azimuthdiff'].data, azidiff)
         self.assertEqual(scene['azimuthdiff'].attrs['name'], 'azimuthdiff')
 
+        np.testing.assert_array_equal(scene['satazimuth'].data, sata)
+        self.assertEqual(scene['satazimuth'].attrs['name'], 'satazimuth')
+
+        np.testing.assert_array_equal(scene['sunazimuth'].data, suna)
+        self.assertEqual(scene['sunazimuth'].attrs['name'], 'sunazimuth')
+
         for angle in ['azimuthdiff', 'satzenith', 'sunzenith']:
             self.assertTupleEqual(scene[angle].dims, ('y', 'x'))
             np.testing.assert_array_equal(scene[angle].coords['x'].data, xvals)