def METHOD_NAME(self):
    mask = np.array(
        [
            [0, 0, 0, 0, 0, 0],
            [0, 0, 1, 1, 0, 0],
            [0, 0, 0, 0, 0, 0],
        ]
    ).astype(np.float32)
    expected_bbox_xywh = [2, 1, 2, 1]
    bbox_xywh = get_bbox_from_mask(mask, 0.5)
    self.assertClose(bbox_xywh, expected_bbox_xywh)