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
|
# (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, Subclass, TraitError, Type
class BaseClass:
pass
class DerivedClass(BaseClass):
pass
class UnrelatedClass:
""" Does not extend the BaseClass """
pass
class ExampleTypeModel(HasTraits):
_class = Type(klass=BaseClass)
class ExampleSubclassModel(HasTraits):
_class = Subclass(BaseClass)
class TypeTestCase(unittest.TestCase):
""" Tests the Type trait and its alias - the Subclass trait"""
def test_type_base(self):
model = ExampleTypeModel(_class=BaseClass)
self.assertIsInstance(model._class(), BaseClass)
def test_type_derived(self):
model = ExampleTypeModel(_class=DerivedClass)
self.assertIsInstance(model._class(), DerivedClass)
def test_invalid_type(self):
example_model = ExampleTypeModel()
def assign_invalid():
example_model._class = UnrelatedClass
self.assertRaises(TraitError, assign_invalid)
def test_subclass_base(self):
model = ExampleSubclassModel(_class=BaseClass)
self.assertIsInstance(model._class(), BaseClass)
def test_subclass_derived(self):
model = ExampleSubclassModel(_class=DerivedClass)
self.assertIsInstance(model._class(), DerivedClass)
def test_invalid_subclass(self):
example_model = ExampleSubclassModel()
def assign_invalid():
example_model._class = UnrelatedClass
self.assertRaises(TraitError, assign_invalid)
|
