Skip to content

Commit 0863a7c

Browse files
committed
Remove a use of getattr for a more type safe alternative
1 parent 4cacab1 commit 0863a7c

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

tests/mock_vws/test_database_summary.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,13 @@ def _wait_for_image_numbers(
8080
raise ValueError(timeout_message)
8181

8282
report = vws_client.get_database_summary_report()
83-
relevant_images_in_summary = getattr(report, key)
83+
key_to_relevant_images_in_summary = {
84+
"active_images": report.active_images,
85+
"inactive_images": report.inactive_images,
86+
"failed_images": report.failed_images,
87+
"processing_images": report.processing_images,
88+
}
89+
relevant_images_in_summary = key_to_relevant_images_in_summary[key]
8490
if value != relevant_images_in_summary: # pragma: no cover
8591
message = (
8692
f"Expected {value} `{key}`s. "

0 commit comments

Comments
 (0)