Fix serialization tests by updating moto import

alteryx · Feb 6, 2024 · 1aed0d5 · 1aed0d5
1 parent 1838f96
commit 1aed0d5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/featuretools/tests/entityset_tests/test_serialization.py b/featuretools/tests/entityset_tests/test_serialization.py
@@ -218,9 +218,9 @@ def test_to_pickle_id_none(tmp_path):
 @pytest.fixture
 def s3_client():
     _environ = os.environ.copy()
-    from moto import mock_s3
+    from moto import mock_aws
 
-    with mock_s3():
+    with mock_aws():
         s3 = boto3.resource("s3")
         yield s3
     os.environ.clear()

diff --git a/featuretools/tests/primitive_tests/test_feature_serialization.py b/featuretools/tests/primitive_tests/test_feature_serialization.py
@@ -210,9 +210,9 @@ def serialize_name_unchanged(original):
 @pytest.fixture
 def s3_client():
     _environ = os.environ.copy()
-    from moto import mock_s3
+    from moto import mock_aws
 
-    with mock_s3():
+    with mock_aws():
         s3 = boto3.resource("s3")
         yield s3
     os.environ.clear()