Use raises as a context manager

Signed-off-by: Ivan Kanakarakis <[email protected]>

Author: c00kiemon5ter

tests/test_02_saml.py

@@ -43,13 +43,15 @@ def test_loadd(self):
         print(ava)
         ee = saml2.ExtensionElement("")
 
-        raises(KeyError, "ee.loadd(ava)")
+        with raises(KeyError):
+            ee.loadd(ava)
 
         ava["tag"] = "foo"
         del ava["namespace"]
 
         ee = saml2.ExtensionElement("")
-        raises(KeyError, "ee.loadd(ava)")
+        with raises(KeyError):
+            ee.loadd(ava)
 
     def test_find_children(self):
         ava = {
@@ -211,8 +213,8 @@ def test_make_vals_str(self):
     def test_make_vals_multi_dict(self):
         ava = ["foo", "bar", "lions", "saints"]
 
-        raises(Exception,
-               "saml2.make_vals(ava, AttributeValue, Attribute(), part=True)")
+        with raises(Exception):
+            saml2.make_vals(ava, AttributeValue, Attribute(), part=True)
 
         attr = Attribute()
         saml2.make_vals(ava, AttributeValue, attr, prop="attribute_value")
@@ -654,7 +656,7 @@ def testUsingTestData(self):
 xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
 NameFormat="urn:oasis:names:tc:SAML:2.0:attrname-format:basic"
 Name="FirstName">
-<AttributeValue 
+<AttributeValue
 xsi:type="xs:base64Binary">VU5JTkVUVA==</AttributeValue>
 </Attribute>"""