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()