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)
