def test_get_distance_buffer_columns_grid_rows(self):
"""Ensures correct output from _get_distance_buffer_columns.
In this case, should return columns with grid rows only.
"""
these_column_names = gridded_forecasts._get_distance_buffer_columns(
EMPTY_STORM_OBJECT_TABLE,
column_type=gridded_forecasts.GRID_ROWS_IN_POLYGON_COLUMN_TYPE)
self.assertTrue(
set(these_column_names) ==
{SMALL_BUFFER_GRID_ROWS_COLUMN, LARGE_BUFFER_GRID_ROWS_COLUMN})
def test_get_distance_buffer_columns_latlng(self):
"""Ensures correct output from _get_distance_buffer_columns.
In this case, should return columns with lat-long polygons only.
"""
these_column_names = gridded_forecasts._get_distance_buffer_columns(
EMPTY_STORM_OBJECT_TABLE,
column_type=gridded_forecasts.LATLNG_POLYGON_COLUMN_TYPE)
self.assertTrue(
set(these_column_names) ==
{SMALL_BUFFER_LATLNG_COLUMN, LARGE_BUFFER_LATLNG_COLUMN})
def test_get_distance_buffer_columns_forecast(self):
"""Ensures correct output from _get_distance_buffer_columns.
In this case, should return columns with forecast probabilities only.
"""
these_column_names = gridded_forecasts._get_distance_buffer_columns(
EMPTY_STORM_OBJECT_TABLE,
column_type=gridded_forecasts.FORECAST_COLUMN_TYPE)
self.assertTrue(
set(these_column_names) ==
{SMALL_BUFFER_FORECAST_COLUMN, LARGE_BUFFER_FORECAST_COLUMN})
def test_get_distance_buffer_columns_xy(self):
"""Ensures correct output from _get_distance_buffer_columns.
In this case, should return columns with x-y polygons only.
"""
these_column_names = gridded_forecasts._get_distance_buffer_columns(
storm_object_table=EMPTY_STORM_OBJECT_TABLE,
column_type=gridded_forecasts.XY_POLYGON_COLUMN_TYPE)
self.assertTrue(
set(these_column_names) ==
{SMALL_BUFFER_XY_COLUMN, LARGE_BUFFER_XY_COLUMN}
)