replace str() with repr() & test it (#1369) (#1473)

stac-utils · Nov 6, 2024 · 5b72d15 · 5b72d15
1 parent b052b2a
commit 5b72d15
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/pystac/utils.py b/pystac/utils.py
@@ -80,7 +80,7 @@ class StringEnum(str, Enum):
     value."""
 
     def __repr__(self) -> str:
-        return str(self.value)
+        return repr(self.value)
 
     def __str__(self) -> str:
         return cast(str, self.value)

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -10,6 +10,7 @@
 from pystac import utils
 from pystac.utils import (
     JoinType,
+    StringEnum,
     is_absolute_href,
     is_file_path,
     join_path_or_url,
@@ -474,3 +475,10 @@ def test_join_path_or_url() -> None:
 )
 def test_is_file_path(href: str, expected: bool) -> None:
     assert is_file_path(href) == expected
+
+
+def test_stringenum_repr() -> None:
+    class SomeEnum(StringEnum):
+        THIS = "this"
+
+    assert repr(SomeEnum.THIS) == "'this'"