Fix tests

Author: hovnatan

tests/test_direct_s3_upload.py

@@ -6,8 +6,6 @@
 
 import superannotate as sa
 
-sa.init(Path.home() / ".superannotate" / "config.json")
-
 TEST_PROJECT_NAME = "test_direct_s3_upload"
 S3_BUCKET = 'superannotate-python-sdk-test'
 S3_FOLDER = 'sample_project_vector'
@@ -42,39 +40,3 @@ def test_direct_s3_upload():
                     on_s3.append(key)
 
     assert len(on_s3) == sa.get_project_image_count(project)
-
-
-@pytest.mark.skipif(
-    "AO_TEST_LEVEL" not in os.environ or
-    os.environ["AO_TEST_LEVEL"] != "stress",
-    reason="Requires env variable to be set"
-)
-def test_direct_s3_upload_stress():
-    projects_found = sa.search_projects(
-        TEST_PROJECT_NAME + " stress", return_metadata=True
-    )
-    for pr in projects_found:
-        sa.delete_project(pr)
-
-    project = sa.create_project(TEST_PROJECT_NAME + " stress", "a", "Vector")
-
-    csv = (Path.home() /
-           "hovnatan_aws.csv").read_text().splitlines()[1].split(",")
-
-    sa.upload_images_from_s3_bucket_to_project(
-        project, csv[2], csv[3], S3_BUCKET, S3_FOLDER_STRESS
-    )
-    s3_client = boto3.client('s3')
-    paginator = s3_client.get_paginator('list_objects_v2')
-    response_iterator = paginator.paginate(
-        Bucket=S3_BUCKET, Prefix=S3_FOLDER_STRESS
-    )
-    on_s3 = []
-    for response in response_iterator:
-        if 'Contents' in response:
-            for object_data in response['Contents']:
-                key = object_data['Key']
-                if key[-4:] == ".jpg":
-                    on_s3.append(key)
-
-    assert len(on_s3) == sa.get_project_image_count(project)

tests/test_team_invite.py

@@ -2,17 +2,14 @@
 
 import superannotate as sa
 
-sa.init(Path.home() / ".superannotate" / "config.json")
-
 
 def test_team_invite():
     try:
-        sa.invite_contributor_to_team("hovnatan@supenotate.com")
+        sa.invite_contributor_to_team("hovnatan@superannotate.com")
     except sa.SABaseException as e:
-        assert e.message == 'Couldn\'t invite to team. {"error":"User with hovnatan@supenotate.com email is already invited to this team."}'
+        assert e.message == 'Couldn\'t invite to team. {"error":"User with hovnatan@superannotate.com email is already member of this team."}'
     else:
         assert False
-
     # sa.delete_team_contributor_invitation(invite)
 
     # print(sa.search_team_contributors())

tests/test_users_and_roles.py

@@ -2,8 +2,6 @@
 
 import superannotate as sa
 
-sa.init(Path.home() / ".superannotate" / "config.json")
-
 PROJECT_NAME = "test users and roles"