tests/test_ml_utils.py

"""
Testing Detection utility functions
"""

import torch
from detection import ml_utils


# Test dictionary conversion
def test_convert_tensor_dict_to_json():
    """Testing tensor to dict conversions"""

    my_dict = {"scores": torch.tensor([1, 2, 3])}
    my_list_gt = {"scores": [1, 2, 3]}
    my_list = ml_utils.convert_tensor_dict_to_json(my_dict)
    assert my_list == my_list_gt