File: test_color.py

package info (click to toggle)
python-enaml 0.19.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 13,284 kB
  • sloc: python: 31,443; cpp: 4,499; makefile: 140; javascript: 68; lisp: 53; sh: 20
file content (56 lines) | stat: -rw-r--r-- 1,493 bytes parent folder | download
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
49
50
51
52
53
54
55
56
#------------------------------------------------------------------------------
# Copyright (c) 2018-2024, Nucleic Development Team.
#
# Distributed under the terms of the Modified BSD License.
#
# The full license is in the file LICENSE, distributed with this software.
#------------------------------------------------------------------------------
import pickle

from enaml.colors import Color


def test_color_initialization():
    """Test the different initialization format supported by Color.

    """
    components = [1, 2, 3, 4]
    c = Color(*components)
    for name, value in zip(['red', 'green', 'blue', 'alpha'], components):
        assert getattr(c, name) == value

    c = Color(alpha=-1)
    for name in ['red', 'green', 'blue', 'alpha']:
        assert getattr(c, name) == 0

    assert c._tkdata is None

    try:
        from enaml.qt.q_resource_helpers import get_cached_qcolor
    except Exception:
        return
    get_cached_qcolor(c)
    assert c._tkdata is not None


def test_color_repr():
    """Test the repr of Color.

    """
    c = Color(1, 2, 3)
    print(repr(c))
    cbis = eval(repr(c))
    for attr in ('red', 'green', 'blue', 'alpha', 'argb'):
        assert getattr(c, attr) == getattr(cbis, attr)


def test_color_pickle():
    """Test pickling/unpickling color.

    """
    rgba = 1, 2, 3, 4
    c = Color(*rgba)
    print(c.__reduce__())
    blob = pickle.dumps(c)
    c = pickle.loads(blob)
    assert (c.red, c.green, c.blue, c.alpha) == rgba