Spaces:
Build error
Build error
File size: 1,223 Bytes
c7e10ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
# Copyright (c) OpenMMLab. All rights reserved.
import pytest
from mmdet.utils import get_caller_name, log_img_scale
def callee_func():
caller_name = get_caller_name()
return caller_name
class CallerClassForTest:
def __init__(self):
self.caller_name = callee_func()
def test_get_caller_name():
# test the case that caller is a function
caller_name = callee_func()
assert caller_name == 'test_get_caller_name'
# test the case that caller is a method in a class
caller_class = CallerClassForTest()
assert caller_class.caller_name == 'CallerClassForTest.__init__'
def test_log_img_scale():
img_scale = (800, 1333)
done_logging = log_img_scale(img_scale)
assert done_logging
img_scale = (1333, 800)
done_logging = log_img_scale(img_scale, shape_order='wh')
assert done_logging
with pytest.raises(ValueError):
img_scale = (1333, 800)
done_logging = log_img_scale(img_scale, shape_order='xywh')
img_scale = (640, 640)
done_logging = log_img_scale(img_scale, skip_square=False)
assert done_logging
img_scale = (640, 640)
done_logging = log_img_scale(img_scale, skip_square=True)
assert not done_logging
|