Skip to content

Commit 508a47a

Browse files
committed
fix protein test for mock open
1 parent 71fa9fe commit 508a47a

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/unit/readers/testProteinDataReader.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ class TestProteinDataReader(unittest.TestCase):
1414

1515
@classmethod
1616
@patch(
17-
"chebai_proteins.preprocessing.reader.open",
17+
"chebai.preprocessing.reader.open",
1818
new_callable=mock_open,
1919
read_data="M\nK\nT\nF\nR\nN",
2020
)
@@ -166,9 +166,7 @@ def test_finish_method_for_new_tokens(self, mock_file: mock_open) -> None:
166166
mock_file.assert_called_with(self.reader.token_path, "a")
167167

168168
# Verify the new tokens were written in the correct order
169-
mock_file().writelines.assert_called_with(
170-
["[H-]\n", "Br\n", "Cl\n", "Na\n", "Mg\n"]
171-
)
169+
mock_file().writelines.assert_called_with(["W\n", "Y\n", "V\n", "Q\n", "E\n"])
172170

173171
def test_finish_method_no_new_tokens(self) -> None:
174172
"""

0 commit comments

Comments
 (0)