Skip to content

Commit 617b4e7

Browse files
committed
fix: tests
1 parent 8f85a7d commit 617b4e7

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

tests/test_datasets.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -487,9 +487,10 @@ def test_get_dataset_run_with_folder_names():
487487

488488
# Fetch the specific run using the new wrapper method
489489
run = langfuse.get_dataset_run(dataset_name=folder_name, run_name=run_name)
490-
assert run.run_name == run_name
490+
assert run.name == run_name
491+
assert run.dataset_name == folder_name
491492
assert run.metadata == {"key": "value"}
492-
assert "/" in folder_name # Verify slashes are preserved
493+
assert "/" in run_name # Verify slashes are preserved in run name
493494

494495

495496
def test_delete_dataset_run_with_folder_names():
@@ -519,7 +520,7 @@ def test_delete_dataset_run_with_folder_names():
519520

520521
# Delete the run using the new wrapper method
521522
result = langfuse.delete_dataset_run(dataset_name=folder_name, run_name=run_name)
522-
assert result.deleted_run_items_count == 1
523+
assert result.message is not None
523524

524525
time.sleep(1) # Give API time to process deletion
525526

0 commit comments

Comments
 (0)