fix to extrapolate for ossm wind

py_gnome/gnome/environment/wind.py

@@ -553,10 +553,13 @@ def at(self, points, time, *, units=None, extrapolate=None, coord_sys='uv',_auto
         else:
             raise ValueError('invalid coordinate system {0}'.format(coord_sys))
 
+        # since extrapolation_is_allowed is not passed into the C code we need
+        # to temporarily set it if extrapolate is passed in, and then change it back
+        original_extrapolation =  self.extrapolation_is_allowed # save value
         if extrapolate is not None: # passed in a value, assume this is what we want
             self.extrapolation_is_allowed = extrapolate
         else: # otherwise probably set on the wind_mover
-            if self.extrapolation_is_allowed is None:
+            if self.extrapolation_is_allowed is None: # shouldn't happen
                 self.extrapolation_is_allowed = False
         #print(f"{self.extrapolation_is_allowed=}")
         try:
@@ -565,6 +568,8 @@ def at(self, points, time, *, units=None, extrapolate=None, coord_sys='uv',_auto
             # CyTimeseries is raising an IndexError
             raise ValueError(f'time specified ({time}) is not within the bounds of the time: '
                              f'({self.data_start} to {self.data_stop})')
+        self.extrapolation_is_allowed = original_extrapolation # put it back the way it was
+
         if idx is None:
             ret_data[:, 0] = data[0]
             ret_data[:, 1] = data[1]

py_gnome/tests/unit_tests/test_environment/test_wind.py

@@ -238,6 +238,10 @@ def test_at_out_of_bounds(wind_circ):
     result = wind.at([(0, 0, 0)], wind.data_stop)
     assert np.array_equal(result, [[1., 0., 0.]])
 
+    # but should work if extrapolate=True
+    result = wind.at([(0, 0, 0)], wind.data_stop + gs.hours(1), extrapolate=True)
+    assert np.array_equal(result, [[1., 0., 0.]])
+
     # after the data_stop
     with pytest.raises(ValueError) as excinfo:
         result = wind.at([(0, 0, 0)], wind.data_stop + gs.hours(1))
@@ -248,10 +252,6 @@ def test_at_out_of_bounds(wind_circ):
         result = wind.at([(0, 0, 0)], wind.data_start - gs.hours(1))
     assert "not within the bounds" in str(excinfo.value)
 
-    # but should work if extrapolate=True
-    result = wind.at([(0, 0, 0)], wind.data_stop + gs.hours(1), extrapolate=True)
-    assert np.array_equal(result, [[1., 0., 0.]])
-
 
 @pytest.fixture(scope='module')
 def wind_rand(rq_rand):