Skip to content

Commit 585eada

Browse files
Test display helper subclass rejection boundaries
Co-authored-by: Shri Sukhani <shrisukhani@users.noreply.github.com>
1 parent 3df331a commit 585eada

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

tests/test_display_utils.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,3 +64,26 @@ def test_format_string_key_for_error_uses_default_fallback_for_invalid_blank_fal
6464
format_string_key_for_error(" ", max_length=20, blank_fallback=123)
6565
== "<blank key>"
6666
)
67+
68+
69+
def test_format_string_key_for_error_rejects_string_subclass_keys():
70+
class _StringSubclass(str):
71+
pass
72+
73+
assert format_string_key_for_error(_StringSubclass("key"), max_length=20) == (
74+
"<blank key>"
75+
)
76+
77+
78+
def test_format_string_key_for_error_rejects_string_subclass_blank_fallbacks():
79+
class _StringSubclass(str):
80+
pass
81+
82+
assert (
83+
format_string_key_for_error(
84+
" ",
85+
max_length=20,
86+
blank_fallback=_StringSubclass("<custom>"),
87+
)
88+
== "<blank key>"
89+
)

0 commit comments

Comments
 (0)