fix flaky test

cognitedata · Nov 17, 2023 · 5e269af · 5e269af
1 parent 223221e
commit 5e269af
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/cognite/client/_api/iam.py b/cognite/client/_api/iam.py
@@ -156,7 +156,7 @@ def compare_capabilities(
         for key, check_grp in to_check_lookup.items():
             group = has_capabilties_lookup.get(key, set())
             if any(AllScope._scope_name == grp[2] for grp in group):
-                continue  # If allScope exists for capability, we safely skip ahead:
+                continue  # If allScope exists for capability, we safely skip ahead
             elif RawAcl._capability_name == next(iter(check_grp))[0]:
                 raw_group.update(group)
                 raw_check_grp.update(check_grp)

diff --git a/tests/tests_unit/test_data_classes/test_capabilities.py b/tests/tests_unit/test_data_classes/test_capabilities.py
@@ -380,10 +380,9 @@ def test_raw_acl_database_scope(self, cognite_client, extra_existing):
         if extra_existing:
             assert missing == [RawAcl([RawAcl.Action.Write], RawAcl.Scope.Table({"db1": ["t1"]}))]
         else:
-            assert missing == [
-                RawAcl([RawAcl.Action.Read], RawAcl.Scope.Table({"db1": ["t3"]})),
-                RawAcl([RawAcl.Action.Write], RawAcl.Scope.Table({"db1": ["t1"]})),
-            ]
+            assert len(missing) == 2
+            assert RawAcl([RawAcl.Action.Read], RawAcl.Scope.Table({"db1": ["t3"]})) in missing
+            assert RawAcl([RawAcl.Action.Write], RawAcl.Scope.Table({"db1": ["t1"]})) in missing
 
 
 @pytest.mark.parametrize(