def test_get_remaining_batch_size_by_class_wind_enough(self):
"""Ensures correct output from _get_remaining_batch_size_by_class.
In this case, target variable is wind and there are enough data in
memory from both classes.
"""
this_dict = trainval_io._get_remaining_batch_size_by_class(
class_to_batch_size_dict=CLASS_TO_BATCH_SIZE_DICT_WIND,
target_values_in_memory=TARGET_VALUES_WIND)
self.assertTrue(this_dict == {-2: 0, 0: 0, 1: 0, 2: 0})
def test_get_remaining_batch_size_by_class_wind_no_data(self):
"""Ensures correct output from _get_remaining_batch_size_by_class.
In this case, target variable is wind and there are no data in
memory.
"""
this_dict = trainval_io._get_remaining_batch_size_by_class(
class_to_batch_size_dict=CLASS_TO_BATCH_SIZE_DICT_WIND,
target_values_in_memory=None)
self.assertTrue(this_dict == CLASS_TO_BATCH_SIZE_DICT_WIND)
