add extra assert

PrefectHQ · Nov 15, 2024 · e07d6c2 · e07d6c2
1 parent b0ae91d
commit e07d6c2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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: