def test_GIVEN_masked_array_WHEN_fix_longitude_THEN_longitudes_fixed_but_masked_values_untouched(
self):
lons = numpy.ma.masked_less([0, -639, 90, 180, -639, 270, 360, -639],
-360)
new_lons = fix_longitude_range(lons, -180)
expected_new_lons = numpy.ma.masked_invalid(
[0, float('Nan'), 90, -180,
float('Nan'), -90, 0,
float('Nan')])
compare_masked_arrays(new_lons, expected_new_lons)
def test_GIVEN_ungridded_missing_values_WHEN_calculate_THEN_missing_values_preserved(self):
data = mock.make_regular_2d_ungridded_data_with_missing_values()
data.metadata._name = 'var1'
self.data = [data]
expr = 'var1 + 10'
res = self.calc.evaluate(self.data, expr)
expected = numpy.ma.masked_invalid([[11, 12, 13], [14, float('Nan'), 16], [17, 18, float('Nan')],
[20, 21, 22], [float('Nan'), 24, 25]]).flatten()
compare_masked_arrays(res.data, expected)
def test_GIVEN_masked_array_WHEN_fix_longitude_THEN_longitudes_fixed_but_masked_values_untouched(self):
lons = numpy.ma.masked_less([0, -639, 90, 180, -639, 270, 360, -639], -360)
new_lons = fix_longitude_range(lons, -180)
expected_new_lons = numpy.ma.masked_invalid([0, float('Nan'), 90, -180, float('Nan'), -90, 0, float('Nan')])
compare_masked_arrays(new_lons, expected_new_lons)