From 5ff75c443ad29c2959faa798601d9834e1082838 Mon Sep 17 00:00:00 2001 From: hagen-danswer Date: Tue, 7 Jan 2025 14:48:03 -0800 Subject: [PATCH] done --- .../daily/connectors/google_drive/consts_and_utils.py | 10 ++++------ .../daily/connectors/google_drive/test_slim_docs.py | 2 +- 2 files changed, 5 insertions(+), 7 deletions(-) diff --git a/backend/tests/daily/connectors/google_drive/consts_and_utils.py b/backend/tests/daily/connectors/google_drive/consts_and_utils.py index e75b44630a8..df829fe743d 100644 --- a/backend/tests/daily/connectors/google_drive/consts_and_utils.py +++ b/backend/tests/daily/connectors/google_drive/consts_and_utils.py @@ -136,15 +136,13 @@ def print_discrepencies( expected: set[str], retrieved: set[str], ) -> None: - # Filter retrieved set to only include valid prefixed items - filtered_retrieved = filter_invalid_prefixes(retrieved) - if expected != filtered_retrieved: + if expected != retrieved: print(expected) - print(filtered_retrieved) + print(retrieved) print("Extra:") - print(filtered_retrieved - expected) + print(retrieved - expected) print("Missing:") - print(expected - filtered_retrieved) + print(expected - retrieved) def _get_expected_file_content(file_id: int) -> str: diff --git a/backend/tests/daily/connectors/google_drive/test_slim_docs.py b/backend/tests/daily/connectors/google_drive/test_slim_docs.py index ffe24d6ba02..1248f6d7363 100644 --- a/backend/tests/daily/connectors/google_drive/test_slim_docs.py +++ b/backend/tests/daily/connectors/google_drive/test_slim_docs.py @@ -85,7 +85,7 @@ def assert_correct_access_for_user( filtered_retrieved_file_names = filter_invalid_prefixes(retrieved_file_names) print_discrepencies(expected_file_names, filtered_retrieved_file_names) - assert expected_file_names == retrieved_file_names + assert expected_file_names == filtered_retrieved_file_names # This function is supposed to map to the group_sync.py file for the google drive connector