tests/test_dig_aws.py

@@ -97,7 +97,7 @@ def test_lookup(test_dig):
 
     with pytest.raises(ValueError) as e:
         test_dig.lookup("what are you talking about")
-    assert str(e.value).startswith("Wrong IP or CIDR format")
+        assert e.startswith("Wrong IP or CIDR format")
 
 
 def test_response_plain_print(test_dig, capsys):

tests/test_get_aws_ip_ranges_json.py

@@ -83,4 +83,4 @@ def test_get_aws_ip_ranges_invalid_status(mocker, fs, create_cache_dir) -> None:
     with pytest.raises(digaws.UnexpectedRequestException) as e:
         digaws.get_aws_ip_ranges()
 
-    assert str(e.value).startswith("Unexpected response from")
+    assert e.match("^Unexpected response from")