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
57
58
59
60
61
62
63
64
65
66
67
|
# (C) Copyright 2004-2023 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!
from unittest import TestCase
from traits.api import HasTraits, Str, Int, List
from traitsui.api import View, Group, Item, TableEditor, ObjectColumn, RGBColor
from traitsui.color_column import ColorColumn
from traitsui.tests._tools import (
BaseTestMixin,
create_ui,
requires_toolkit,
reraise_exceptions,
ToolkitName,
)
class MyEntry(HasTraits):
name = Str()
value = Int(0)
color = RGBColor()
entry_view = View(Group(Item("name"), Item("value"), Item("color")))
my_editor = TableEditor(
columns=[
ObjectColumn(name="name"),
ObjectColumn(name="value"),
ColorColumn(name="color", style="readonly"),
],
orientation="vertical",
show_toolbar=True,
row_factory=MyEntry,
)
class MyData(HasTraits):
data_list = List(MyEntry)
view = View(Item("data_list", editor=my_editor, show_label=False))
class TestColorColumn(BaseTestMixin, TestCase):
def setUp(self):
BaseTestMixin.setUp(self)
def tearDown(self):
BaseTestMixin.tearDown(self)
@requires_toolkit([ToolkitName.qt, ToolkitName.wx])
def test_color_column(self):
# Behaviour: column ui should display without error
d1 = MyEntry(name="a", value=2, color=(1.0, 0.3, 0.1))
d2 = MyEntry(name="b", value=3, color=(0.1, 0.0, 0.9))
data = MyData(data_list=[d1, d2])
with reraise_exceptions(), create_ui(data):
pass
|
