def test_max_num_query_respected(self):
images_per_class = np.array([30, 45, 35, 50])
num_query = sampling.compute_num_query(
images_per_class,
max_num_query=test_utils.MAX_NUM_QUERY,
num_support=None)
self.assertEqual(num_query, test_utils.MAX_NUM_QUERY)
def test_at_most_half(self):
images_per_class = np.array([10, 9, 20, 21])
num_query = sampling.compute_num_query(
images_per_class,
max_num_query=test_utils.MAX_NUM_QUERY,
num_support=None)
self.assertEqual(num_query, 4)
def test_raises_error_on_one_image_per_class(self):
images_per_class = np.array([1, 3, 8, 8])
with self.assertRaises(ValueError):
sampling.compute_num_query(images_per_class,
max_num_query=test_utils.MAX_NUM_QUERY)
def test_max_num_query_respected(self):
images_per_class = np.array([30, 45, 35, 50])
num_query = sampling.compute_num_query(
images_per_class, max_num_query=MAX_NUM_QUERY)
self.assertEqual(num_query, MAX_NUM_QUERY)
