AMMICO/misinformation/test/test_display.py
2023-02-15 03:31:39 -08:00

28 строки
890 B
Python

import json
# import misinformation.display as misinf_display
import pytest
misinf_display = pytest.importorskip("misinformation.display")
def test_explore_analysis_faces():
mydict = {"IMG_2746": {"filename": "./test/data/IMG_2746.png"}}
misinf_display.explore_analysis(mydict, identify="faces")
with open("./test/data/example_faces.json", "r") as file:
outs = json.load(file)
for im_key in mydict.keys():
sub_dict = mydict[im_key]
for key in sub_dict.keys():
assert sub_dict[key] == outs[key]
def test_explore_analysis_objects():
mydict = {"IMG_2746": {"filename": "./test/data/IMG_2809.png"}}
misinf_display.explore_analysis(mydict, identify="objects")
with open("./test/data/example_analysis_objects.json", "r") as file:
outs = json.load(file)
assert str(mydict) == str(outs)