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)