Modified test with updated behaviour of API

OpenMined · Aug 6, 2024 · 710ceb1 · 710ceb1
1 parent 2cb44bf
commit 710ceb1
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/packages/syft/tests/syft/api_test.py b/packages/syft/tests/syft/api_test.py
@@ -3,11 +3,10 @@
 
 # third party
 import numpy as np
-import pytest
 
 # syft absolute
 import syft as sy
-from syft.service.response import SyftAttributeError
+from syft.service.response import SyftError
 from syft.service.user.user_roles import ServiceRole
 
 
@@ -58,8 +57,7 @@ def get_role(verify_key):
     dataset = sy.Dataset(
         name="test2",
     )
-    with pytest.raises(SyftAttributeError):
-        assert guest_client.upload_dataset(dataset)
+    assert isinstance(guest_client.upload_dataset(dataset), SyftError)
 
     assert guest_client.api.services.user.update(uid=user_id, name="abcdef")