Skip to content

Commit

Permalink
Replace with a test that at least exercises the create_editor method
Browse files Browse the repository at this point in the history
  • Loading branch information
mdickinson committed Apr 24, 2024
1 parent 840a824 commit eec04bc
Showing 1 changed file with 15 additions and 7 deletions.
22 changes: 15 additions & 7 deletions traits/tests/test_enum.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
from traits.api import (
Any, BaseEnum, Enum, HasTraits, Int, List, Property, Set, TraitError,
Tuple)
from traits.testing.optional_dependencies import requires_traitsui


class FooEnum(enum.Enum):
Expand Down Expand Up @@ -111,13 +112,6 @@ class EnumCollectionExample(HasTraits):
slow_enum = BaseEnum("yes", "no", "maybe")


class EnumCollectionGUIExample(EnumCollectionExample):
# Override attributes that may fail GUI test
# until traitsui #781 is fixed.
int_set_enum = Enum("int", "set")
correct_int_set_enum = Enum("int", "set")


class EnumTestCase(unittest.TestCase):
def test_valid_enum(self):
example_model = ExampleModel(root="model1")
Expand Down Expand Up @@ -336,3 +330,17 @@ class HasEnumInList(HasTraits):
model.digit_sequence = [-1, 0, 1, 1]
with self.assertRaises(TraitError):
model.digit_sequence = [-1, 0, 2, 1]


@requires_traitsui
class TestEnumCreateEditor(unittest.TestCase):
def test_create_editor(self):
import traitsui.editor_factory

obj = EnumListExample()
trait = obj.trait("value")
editor_factory = trait.trait_type.create_editor()
self.assertIsInstance(
editor_factory,
traitsui.editor_factory.EditorFactory
)

0 comments on commit eec04bc

Please sign in to comment.