Spaces:

cvachet
/

object_detection_lambda

Sleeping

object_detection_lambda / tests /test_ml_utils.py

Clement Vachet

Add unit tests via pytest

b41850c almost 2 years ago

309 Bytes

from detection import ml_utils
import torch
import json


# Test dictionary conversion
def test_convert_tensor_dict_to_json():
    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

	from detection import ml_utils
	import torch
	import json


	# Test dictionary conversion
	def test_convert_tensor_dict_to_json():
	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