def test_column_name_to_buffer_small_buffer_grid_rows(self): """Ensures correct output from _column_name_to_buffer. In this case, the buffer is "inside storm" and the column name is for grid rows inside the polygon. """ this_min_distance_metres, this_max_distance_metres = ( gridded_forecasts._column_name_to_buffer( SMALL_BUFFER_GRID_ROWS_COLUMN) ) self.assertTrue(numpy.isclose( this_min_distance_metres, SMALL_BUFFER_MIN_DISTANCE_METRES, equal_nan=True, atol=TOLERANCE )) self.assertTrue(numpy.isclose( this_max_distance_metres, SMALL_BUFFER_MAX_DISTANCE_METRES, atol=TOLERANCE ))
def test_column_name_to_buffer_large_buffer_grid_columns(self): """Ensures correct output from _column_name_to_buffer. In this case, the buffer is "5-10 km outside storm" and the column name is for grid columns inside the polygon. """ this_min_distance_metres, this_max_distance_metres = ( gridded_forecasts._column_name_to_buffer( LARGE_BUFFER_GRID_COLUMNS_COLUMN) ) self.assertTrue(numpy.isclose( this_min_distance_metres, LARGE_BUFFER_MIN_DISTANCE_METRES, atol=TOLERANCE )) self.assertTrue(numpy.isclose( this_max_distance_metres, LARGE_BUFFER_MAX_DISTANCE_METRES, atol=TOLERANCE ))
def test_column_name_to_buffer_small_buffer_forecast(self): """Ensures correct output from _column_name_to_buffer. In this case, the buffer is "inside storm" and the column name is for forecast probabilities. """ this_min_distance_metres, this_max_distance_metres = ( gridded_forecasts._column_name_to_buffer( SMALL_BUFFER_FORECAST_COLUMN) ) self.assertTrue(numpy.isclose( this_min_distance_metres, SMALL_BUFFER_MIN_DISTANCE_METRES, equal_nan=True, atol=TOLERANCE )) self.assertTrue(numpy.isclose( this_max_distance_metres, SMALL_BUFFER_MAX_DISTANCE_METRES, atol=TOLERANCE ))