"vscode:/vscode.git/clone" did not exist on "063c5489b349fe2d7f786c8196acc2bae5b24ce6"
test_visualization.py 602 Bytes
Newer Older
1
# Copyright (c) OpenMMLab. All rights reserved.
2
3
4
import numpy as np
import pytest

Kai Chen's avatar
Kai Chen committed
5
6
import mmcv

7
8
9
10
11
12
13
14
15
16
17
18
19

def test_color():
    assert mmcv.color_val(mmcv.Color.blue) == (255, 0, 0)
    assert mmcv.color_val('green') == (0, 255, 0)
    assert mmcv.color_val((1, 2, 3)) == (1, 2, 3)
    assert mmcv.color_val(100) == (100, 100, 100)
    assert mmcv.color_val(np.zeros(3, dtype=np.int)) == (0, 0, 0)
    with pytest.raises(TypeError):
        mmcv.color_val([255, 255, 255])
    with pytest.raises(TypeError):
        mmcv.color_val(1.0)
    with pytest.raises(AssertionError):
        mmcv.color_val((0, 0, 500))