diff --git a/tests/test_cloudfront/test_server.py b/tests/test_cloudfront/test_server.py index 0cee4ea5b..534ca606a 100644 --- a/tests/test_cloudfront/test_server.py +++ b/tests/test_cloudfront/test_server.py @@ -1,4 +1,3 @@ -import sure # noqa # pylint: disable=unused-import import xmltodict from moto import mock_cloudfront @@ -12,5 +11,5 @@ def test_cloudfront_list(): res = test_client.get("/2020-05-31/distribution") data = xmltodict.parse(res.data, dict_constructor=dict) - data.should.have.key("DistributionList") - data["DistributionList"].shouldnt.have.key("Items") + assert "DistributionList" in data + assert "Items" not in data["DistributionList"]