Skip to content

Commit 225b68c

Browse files
committed
Fix formatting in context_list assertion for PartialClass in TestCodeContextStructure
1 parent b5740cf commit 225b68c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/core/test_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -418,7 +418,7 @@ def test_as_list_str(self, mock_wrap_content):
418418
assert 'wrapped(import numpy as np, PACKAGES)' in context_list
419419
assert 'wrapped(\nclass FullClass:\n ..., file1.py)' in context_list
420420
# Check that the partial class was constructed correctly
421-
assert 'wrapped(class PartialClass(Base):\n\n\n ...\n\n\ndef partial_method(self):\n pass, file2.py)' in context_list
421+
assert 'wrapped(\nclass PartialClass(Base):\n\n ...\n\ndef partial_method(self):\n pass, file2.py)' in context_list
422422

423423
# Assertions for target list
424424
assert len(target_list) == 2 # 1 preloaded + 1 requested

0 commit comments

Comments
 (0)