STY: simplify

pysat · Dec 23, 2021 · 7af5d65 · 7af5d65
1 parent 6b62c1b
commit 7af5d65
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pysatMissions/tests/test_instruments.py b/pysatMissions/tests/test_instruments.py
@@ -123,9 +123,8 @@ def test_sgp4_data_continuity(self, kwargs):
         gradient_between_days = day2.iloc[0] - day1.iloc[-1]
 
         # Check that the jump between days is within 3 sigma of average gradient
-        for key in average_gradient.keys():
-            del_g = np.abs(average_gradient[key] - gradient_between_days[key])
-            assert del_g < 3. * std_gradient[key]
+        del_g = np.abs(average_gradient - gradient_between_days)
+        assert np.all(del_g < (3. * std_gradient))
 
         return