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
|
# (C) Copyright 2005-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.api import (
HasTraits,
Int,
List,
Str,
TraitError,
)
class Foo(HasTraits):
num = Int
bar = Str
class Bar(HasTraits):
# Default is visible.
PubT1 = Str
# Change to invisible.
PubT2 = Str(visible=False)
# New behaviour makes private traits invisible.
PrivT1 = Str(private=True)
# Force visibility of a private trait.
PrivT2 = Str(private=True, visible=True)
class FooBar(HasTraits):
num = Int
baz = "non-trait class attribute"
class GetTraitTestCase(unittest.TestCase):
def test_trait_set_bad(self):
b = Foo(num=23)
# This should fail before and after #234.
with self.assertRaises(TraitError):
b.num = "first"
self.assertEqual(b.num, 23)
def test_trait_set_replaced(self):
b = Foo()
# Overriding the trait with a new type should work.
b.add_trait("num", Str())
b.num = "first"
self.assertEqual(b.num, "first")
def test_trait_set_replaced_and_check(self):
b = Foo()
b.add_trait("num", Str())
b.num = "first"
self.assertEqual(b.num, "first")
# Check that the "traits" call picks up the new instance trait. (See
# #234.)
self.assertEqual(b.trait("num"), b.traits()["num"])
def test_trait_names_returned_by_visible_traits(self):
b = Bar()
self.assertEqual(
sorted(b.visible_traits()), sorted(["PubT1", "PrivT2"])
)
def test_dir(self):
b = FooBar()
names = dir(b)
self.assertIn("baz", names)
self.assertIn("num", names)
self.assertIn("edit_traits", names)
# Issue 925: _notifiers not shown in dir()
self.assertIn("_notifiers", names)
# Ensure no duplicates
self.assertEqual(len(set(names)), len(names))
def test_trait_name_with_list_items(self):
# Dynamically added a List trait causes an "_items" to be added, but
# that should not be reported by HasTraits.traits
class Base(HasTraits):
pass
a = Base()
# This adds an instance trait for "pins_items" as well.
a.add_trait("pins", List())
self.assertIn("pins", a.traits())
self.assertNotIn("pins_items", a.traits())
def test_trait_name_with_items(self):
# Dynamically added traits with name "*_items" is also skipped.
class Base(HasTraits):
pass
a = Base()
a.add_trait("good_items", Str())
# If enthought/traits#1329 is fixed, then the following assertion
# should fail.
self.assertNotIn("good_items", a.traits())
|
