зеркало из
https://github.com/ssciwr/AMMICO.git
synced 2025-10-30 21:46:04 +02:00
27732c5fab
* add py test * update dependencies and markers * Update ci.yml * fix linting * trying with editable install for cov * fix code smells * fix code smells Co-authored-by: xianghe ma <maxianghe@outlook.com>
34 строки
1.2 KiB
Python
34 строки
1.2 KiB
Python
import json
|
|
from misinformation.display import explore_analysis
|
|
from pytest import approx
|
|
|
|
|
|
def test_explore_analysis_faces():
|
|
mydict = {"IMG_2746": {"filename": "./test/data/IMG_2746.png"}}
|
|
explore_analysis(mydict, identify="faces")
|
|
with open("./test/data/example_analysis_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():
|
|
if key != "emotion":
|
|
assert sub_dict[key] == outs[im_key][key]
|
|
# json can't handle tuples natively
|
|
for i in range(0, len(sub_dict["emotion"])):
|
|
temp = (
|
|
list(sub_dict["emotion"][i])
|
|
if type(sub_dict["emotion"][i]) == tuple
|
|
else sub_dict["emotion"][i]
|
|
)
|
|
assert approx(temp) == outs[im_key]["emotion"][i]
|
|
|
|
|
|
def test_explore_analysis_objects():
|
|
mydict = {"IMG_2746": {"filename": "./test/data/IMG_2809.png"}}
|
|
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)
|