Skip to content

Commit 5c1c8b6

Browse files
committed
Fix tests
1 parent 91efe7e commit 5c1c8b6

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/unit/utils/test_file_handler.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,7 @@ def test_file_handler_write_json_from_write_method_happy_path(
257257
) -> None:
258258
handler = FileHandler()
259259
handler.write("data", "file.json")
260-
mock_write_json.assert_called_with("data", Path("file.json"), overwrite=False)
260+
mock_write_json.assert_called_with("data", Path("file.json"))
261261

262262

263263
@patch("neo4j_graphrag.utils.file_handler.FileHandler._write_yaml")
@@ -268,7 +268,7 @@ def test_file_handler_write_yaml_from_write_method_happy_path(
268268
) -> None:
269269
handler = FileHandler()
270270
handler.write("data", "file.yaml")
271-
mock_write_yaml.assert_called_with("data", Path("file.yaml"), overwrite=False)
271+
mock_write_yaml.assert_called_with("data", Path("file.yaml"))
272272

273273

274274
@patch("neo4j_graphrag.utils.file_handler.FileHandler._write_yaml")
@@ -279,7 +279,7 @@ def test_file_handler_write_yaml_from_write_method_overwrite_format_happy_path(
279279
) -> None:
280280
handler = FileHandler()
281281
handler.write("data", "file.txt", format=FileFormat.YAML)
282-
mock_write_yaml.assert_called_with("data", Path("file.txt"), overwrite=False)
282+
mock_write_yaml.assert_called_with("data", Path("file.txt"))
283283

284284

285285
@patch("neo4j_graphrag.utils.file_handler.FileHandler._write_json")
@@ -290,4 +290,4 @@ def test_file_handler_write_json_from_write_method_overwrite_format_happy_path(
290290
) -> None:
291291
handler = FileHandler()
292292
handler.write("data", "file.txt", format=FileFormat.JSON)
293-
mock_write_json.assert_called_with("data", Path("file.txt"), overwrite=False)
293+
mock_write_json.assert_called_with("data", Path("file.txt"))

0 commit comments

Comments
 (0)