@@ -257,7 +257,7 @@ def test_file_handler_write_json_from_write_method_happy_path(
257
257
) -> None :
258
258
handler = FileHandler ()
259
259
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" ))
261
261
262
262
263
263
@patch ("neo4j_graphrag.utils.file_handler.FileHandler._write_yaml" )
@@ -268,7 +268,7 @@ def test_file_handler_write_yaml_from_write_method_happy_path(
268
268
) -> None :
269
269
handler = FileHandler ()
270
270
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" ))
272
272
273
273
274
274
@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(
279
279
) -> None :
280
280
handler = FileHandler ()
281
281
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" ))
283
283
284
284
285
285
@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(
290
290
) -> None :
291
291
handler = FileHandler ()
292
292
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