first comit

venv/lib/python3.10/site-packages/traitlets/tests/test_traitlets.py

@@ -0,0 +1,59 @@
+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]