diff --git a/tests/test_utils_elog.py b/tests/test_utils_elog.py index b0791c44..e374a141 100644 --- a/tests/test_utils_elog.py +++ b/tests/test_utils_elog.py @@ -120,6 +120,6 @@ def test_screenshot(mock_screenshot_class): message, attributes, attachments = elog.read(msg_id) assert test_msg in message - assert filename in [os.path.basename(a) for a in attachments] + assert any(a.endswith(filename) for a in [os.path.basename(att) for att in attachments]) os.remove(fake_path) \ No newline at end of file