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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
# (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!
import unittest
from traits.has_traits import HasTraits
from traits.trait_types import Float, List, Instance
from traitsui.handler import ModelView
from traitsui.view import View
from traitsui.item import Item
from traitsui.editors.csv_list_editor import CSVListEditor
import traitsui.editors.csv_list_editor as csv_list_editor
from traitsui.tests._tools import (
BaseTestMixin,
create_ui,
is_wx,
is_qt,
requires_toolkit,
reraise_exceptions,
ToolkitName,
)
class ListOfFloats(HasTraits):
data = List(Float)
class ListOfFloatsWithCSVEditor(ModelView):
model = Instance(ListOfFloats)
traits_view = View(
Item(label="Close the window to append data"),
Item("model.data", editor=CSVListEditor()),
buttons=["OK"],
)
class TestCSVEditor(BaseTestMixin, unittest.TestCase):
def setUp(self):
BaseTestMixin.setUp(self)
def tearDown(self):
BaseTestMixin.tearDown(self)
@requires_toolkit([ToolkitName.qt, ToolkitName.wx])
def test_csv_editor_disposal(self):
# Bug: CSVListEditor does not un-hook the traits notifications after
# its disposal, causing errors when the hooked data is accessed after
# the window is closed (Issue #48)
list_of_floats = ListOfFloats(data=[1, 2, 3])
csv_view = ListOfFloatsWithCSVEditor(model=list_of_floats)
try:
with reraise_exceptions():
with create_ui(csv_view):
pass
# raise an exception if still hooked
list_of_floats.data.append(2)
except AttributeError:
# if all went well, we should not be here
self.fail("AttributeError raised")
@requires_toolkit([ToolkitName.qt, ToolkitName.wx])
def test_csv_editor_external_append(self):
# Behavior: CSV editor is notified when an element is appended to the
# list externally
def _wx_get_text_value(ui):
txt_ctrl = ui.control.FindWindowByName("text", ui.control)
return txt_ctrl.GetValue()
def _qt_get_text_value(ui):
from pyface import qt
txt_ctrl = ui.control.findChild(qt.QtGui.QLineEdit)
return txt_ctrl.text()
list_of_floats = ListOfFloats(data=[1.0])
csv_view = ListOfFloatsWithCSVEditor(model=list_of_floats)
with reraise_exceptions(), create_ui(csv_view) as ui:
# add element to list, make sure that editor knows about it
list_of_floats.data.append(3.14)
# get current value from editor
if is_wx():
value_str = _wx_get_text_value(ui)
elif is_qt():
value_str = _qt_get_text_value(ui)
expected = csv_list_editor._format_list_str([1.0, 3.14])
self.assertEqual(value_str, expected)
if __name__ == "__main__":
# Executing the file opens the dialog for manual testing
list_of_floats = ListOfFloats(data=[1, 2, 3])
csv_view = ListOfFloatsWithCSVEditor(model=list_of_floats)
csv_view.configure_traits()
# this call will raise an AttributeError in commit
# 4ecb2fa8f0ef385d55a2a4062d821b0415777973
# This is because the editor does not un-hook the traits notifications
# after its disposal
list_of_floats.data.append(2)
print(list_of_floats.data)
|
