-
Notifications
You must be signed in to change notification settings - Fork 347
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
fix: loosen assertions for system test featurestore #1040
fix: loosen assertions for system test featurestore #1040
Conversation
@@ -79,7 +76,7 @@ def test_create_get_list_featurestore(self, shared_state): | |||
assert featurestore.resource_name == get_featurestore.resource_name | |||
|
|||
list_featurestores = aiplatform.Featurestore.list() | |||
assert (len(list_featurestores) - base_list_featurestores) == 1 | |||
assert len(list_featurestores) >= 1 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
For my system test, I actually do:
assert get_index.resource_name in [
index.resource_name for index in list_indexes
]
Just another option that doesn't depend on the absolute number of items in the list.
@@ -121,7 +118,7 @@ def test_create_get_list_entity_types(self, shared_state): | |||
list_entity_types = aiplatform.EntityType.list( | |||
featurestore_name=featurestore_name | |||
) | |||
assert len(list_entity_types) == 2 | |||
assert len(list_entity_types) >= 1 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
ditto
@@ -179,7 +173,7 @@ def test_create_get_list_features(self, shared_state): | |||
) | |||
|
|||
list_user_features = user_entity_type.list_features() | |||
assert len(list_user_features) == 3 | |||
assert len(list_user_features) >= 1 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
ditto
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Fix: #1030