def test_show_annotation(record, monkeypatch):
monkeypatch.setattr(plt, "show", lambda: None)
data = default_prepare_record(record)
show_annotation(data["img"],
labels=data["label"],
bboxes=data["bbox"],
masks=data["mask"])
plt.show()
def test_show_record_label(record, monkeypatch):
monkeypatch.setattr(plt, "show", lambda: None)
record = record.copy()
record.pop("bboxes")
record.pop("masks")
with pytest.raises(ValueError) as e:
show_record(record)
assert str(
e.value) == "Can only display labels if bboxes or masks are given"
plt.show()
def test_show_record_label_bbox_mask(record, monkeypatch):
monkeypatch.setattr(plt, "show", lambda: None)
show_record(record)
plt.show()
def test_show_record_label_mask(record, monkeypatch):
monkeypatch.setattr(plt, "show", lambda: None)
record = record.copy()
record.pop("bboxes")
show_record(record)
plt.show()
