def METHOD_NAME(self):
    adapter = DetectionOutputAdapter(CXCYWH_LABELS_SCORES_DISTANCE_ATTR, ATTR_YXYX, image_shape=(640, 640)).eval()
    example_inputs = (
        torch.randn((300, CXCYWH_LABELS_SCORES_DISTANCE_ATTR.num_channels)),
        torch.randn((4, 300, CXCYWH_LABELS_SCORES_DISTANCE_ATTR.num_channels)),
    )
    for expected_input in example_inputs:
        intermediate = adapter(expected_input)
        self.assertEqual(ATTR_YXYX.num_channels, intermediate.size(-1))