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)
