def __call__(self, data): d = dict(data) for key in self.keys: bbox = generate_spatial_bounding_box(d[key]) if d.get(self.result) is None: d[self.result] = dict() d[self.result][self.bbox] = np.array(bbox).astype(int).tolist() return d
def test_value(self, input_data, expected_box): result = generate_spatial_bounding_box(**input_data) self.assertTupleEqual(result, expected_box)