60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import Any
|
||
|
|
from unittest import TestCase
|
||
|
|
|
||
|
|
from traitlets import TraitError
|
||
|
|
|
||
|
|
|
||
|
|
class TraitTestBase(TestCase):
|
||
|
|
"""A best testing class for basic trait types."""
|
||
|
|
|
||
|
|
def assign(self, value: Any) -> None:
|
||
|
|
self.obj.value = value # type:ignore[attr-defined]
|
||
|
|
|
||
|
|
def coerce(self, value: Any) -> Any:
|
||
|
|
return value
|
||
|
|
|
||
|
|
def test_good_values(self) -> None:
|
||
|
|
if hasattr(self, "_good_values"):
|
||
|
|
for value in self._good_values:
|
||
|
|
self.assign(value)
|
||
|
|
self.assertEqual(self.obj.value, self.coerce(value)) # type:ignore[attr-defined]
|
||
|
|
|
||
|
|
def test_bad_values(self) -> None:
|
||
|
|
if hasattr(self, "_bad_values"):
|
||
|
|
for value in self._bad_values:
|
||
|
|
try:
|
||
|
|
self.assertRaises(TraitError, self.assign, value)
|
||
|
|
except AssertionError:
|
||
|
|
raise AssertionError(value) from None
|
||
|
|
|
||
|
|
def test_default_value(self) -> None:
|
||
|
|
if hasattr(self, "_default_value"):
|
||
|
|
self.assertEqual(self._default_value, self.obj.value) # type:ignore[attr-defined]
|
||
|
|
|
||
|
|
def test_allow_none(self) -> None:
|
||
|
|
if (
|
||
|
|
hasattr(self, "_bad_values")
|
||
|
|
and hasattr(self, "_good_values")
|
||
|
|
and None in self._bad_values
|
||
|
|
):
|
||
|
|
trait = self.obj.traits()["value"] # type:ignore[attr-defined]
|
||
|
|
try:
|
||
|
|
trait.allow_none = True
|
||
|
|
self._bad_values.remove(None)
|
||
|
|
# skip coerce. Allow None casts None to None.
|
||
|
|
self.assign(None)
|
||
|
|
self.assertEqual(self.obj.value, None) # type:ignore[attr-defined]
|
||
|
|
self.test_good_values()
|
||
|
|
self.test_bad_values()
|
||
|
|
finally:
|
||
|
|
# tear down
|
||
|
|
trait.allow_none = False
|
||
|
|
self._bad_values.append(None)
|
||
|
|
|
||
|
|
def tearDown(self) -> None:
|
||
|
|
# restore default value after tests, if set
|
||
|
|
if hasattr(self, "_default_value"):
|
||
|
|
self.obj.value = self._default_value # type:ignore[attr-defined]
|