Update test/test_traversal.py

Co-authored-by: bashonly <88596187+bashonly@users.noreply.github.com>
pull/13993/head
garret1317 1 month ago committed by GitHub
parent 5b3dbcfa49
commit 7ffe9009c7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -442,13 +442,13 @@ class TestTraversal:
def test_traversal_dataclass(self):
@dataclasses.dataclass
class _TEST_DATACLASS:
string: str
class _TestDataclass:
val: str
dataclass = _TEST_DATACLASS(string='yt-dlp')
assert traverse_obj(dataclass, ('string')) == 'yt-dlp', \
dc = _TestDataclass(val='yt-dlp')
assert traverse_obj(dc, 'val') == 'yt-dlp', \
'Dataclasses should be traversable'
assert traverse_obj({'dataclass': dataclass}, ('dataclass', 'string')) == 'yt-dlp', \
assert traverse_obj({'dataclass': dc}, ('dataclass', 'val')) == 'yt-dlp', \
'Dataclasses inside other objects should be traversable'

Loading…
Cancel
Save