freesurfer-surface/tests/test_label.py

import pytest

from freesurfer_surface import Label


@pytest.mark.parametrize(('red', 'green', 'blue', 'transparency', 'color_code'), [
    (220,  20,  20,   0,  1316060),
    ( 60,  20, 220,   0, 14423100),
    ( 75,  50, 125,   0,  8204875),
    ( 20, 220, 160,   0, 10542100),
])
def test_color_code(red, green, blue, transparency, color_code):
    label = Label(index=21, name='name', red=red, green=green,
                  blue=blue, transparency=transparency)
    assert color_code == label.color_code


def test_color_code_unknown():
    label = Label(index=0, name='unknown', red=21,
                  green=21, blue=21, transparency=0)
    assert label.color_code == 0


@pytest.mark.parametrize(('red', 'green', 'blue', 'hex_color_code'), [
    (  0,   0,   0, '#000000'),
    (255, 255, 255, '#ffffff'),
    (255,   0,   0, '#ff0000'),
    (  0, 255,   0, '#00ff00'),
    (  0,   0, 255, '#0000ff'),
    (  1,   2,   3, '#010203'),
    ( 17,  18,  19, '#111213'),
    (128, 192, 255, '#80c0ff'),
    ( 20, 220, 160, '#14dca0'),
])
def test_hex_color_code(red, green, blue, hex_color_code):
    label = Label(index=21, name='name', red=red,
                  green=green, blue=blue, transparency=0)
    assert hex_color_code == label.hex_color_code.lower()


def test_str():
    label = Label(index=24, name='precentral', red=60,
                  green=20, blue=220, transparency=0)
    assert str(label) == 'Label(name=precentral, index=24, color=#3c14dc)'


def test_repr():
    label = Label(index=24, name='precentral', red=60,
                  green=20, blue=220, transparency=0)
    assert repr(label) == 'Label(name=precentral, index=24, color=#3c14dc)'