def test_distance_buffer_to_column_name_large_buffer_grid_columns(self): """Ensures correct output from _distance_buffer_to_column_name. In this case, the buffer is "5-10 km outside storm" and the column name is for grid columns inside the polygon. """ this_column_name = gridded_forecasts._distance_buffer_to_column_name( LARGE_BUFFER_MIN_DISTANCE_METRES, LARGE_BUFFER_MAX_DISTANCE_METRES, column_type=gridded_forecasts.GRID_COLUMNS_IN_POLYGON_COLUMN_TYPE) self.assertTrue(this_column_name == LARGE_BUFFER_GRID_COLUMNS_COLUMN)
def test_distance_buffer_to_column_name_large_buffer_latlng(self): """Ensures correct output from _distance_buffer_to_column_name. In this case, the buffer is "5-10 km outside storm" and the column name is for lat-long polygons. """ this_column_name = gridded_forecasts._distance_buffer_to_column_name( LARGE_BUFFER_MIN_DISTANCE_METRES, LARGE_BUFFER_MAX_DISTANCE_METRES, column_type=gridded_forecasts.LATLNG_POLYGON_COLUMN_TYPE) self.assertTrue(this_column_name == LARGE_BUFFER_LATLNG_COLUMN)
def test_distance_buffer_to_column_name_large_buffer_forecast(self): """Ensures correct output from _distance_buffer_to_column_name. In this case, the buffer is "5-10 km outside storm" and the column name is for forecast probabilities. """ this_column_name = gridded_forecasts._distance_buffer_to_column_name( LARGE_BUFFER_MIN_DISTANCE_METRES, LARGE_BUFFER_MAX_DISTANCE_METRES, column_type=gridded_forecasts.FORECAST_COLUMN_TYPE) self.assertTrue(this_column_name == LARGE_BUFFER_FORECAST_COLUMN)
def test_distance_buffer_to_column_name_small_buffer_grid_rows(self): """Ensures correct output from _distance_buffer_to_column_name. In this case, the buffer is "inside storm" and the column name is for grid rows inside the polygon. """ this_column_name = gridded_forecasts._distance_buffer_to_column_name( SMALL_BUFFER_MIN_DISTANCE_METRES, SMALL_BUFFER_MAX_DISTANCE_METRES, column_type=gridded_forecasts.GRID_ROWS_IN_POLYGON_COLUMN_TYPE) self.assertTrue(this_column_name == SMALL_BUFFER_GRID_ROWS_COLUMN)