From e07d6c28e00010cbdc24fe3dc06593a97348825e Mon Sep 17 00:00:00 2001 From: zzstoatzz Date: Fri, 15 Nov 2024 12:21:51 -0600 Subject: [PATCH] add extra assert --- flows/ray_result_storage_location.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/flows/ray_result_storage_location.py b/flows/ray_result_storage_location.py index 34e13d733221..262a15a53583 100644 --- a/flows/ray_result_storage_location.py +++ b/flows/ray_result_storage_location.py @@ -61,6 +61,9 @@ def my_flow(): f"Expected exactly two files in {tmp_path}, " f"found {len(found_files)}: {found_files!r}" ) + assert any( + f.name == "test-key" for f in found_files + ), f"Expected to find 'test-key' in {found_files!r}" print("Result storage location test passed!") finally: