From ee69b04b53c2b2b65c9a0dc1d6ae08a98cf9aa26 Mon Sep 17 00:00:00 2001
From: Valentin Samir <valentin.samir@crans.org>
Date: Sun, 29 Apr 2018 19:04:44 +0200
Subject: [PATCH] Fix code style of previous commit

---
 cas_server/tests/mixin.py     | 8 +++++---
 cas_server/tests/test_view.py | 4 +++-
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/cas_server/tests/mixin.py b/cas_server/tests/mixin.py
index e6930a7..48d51f6 100644
--- a/cas_server/tests/mixin.py
+++ b/cas_server/tests/mixin.py
@@ -149,12 +149,14 @@ class XmlContent(object):
             namespaces={'cas': "http://www.yale.edu/tp/cas"}
         )
         self.assertEqual(len(attributes), 1)
-        ignore_attrs = {"authenticationDate", "longTermAuthenticationRequestTokenUsed", "isFromNewLogin"}
+        ignore_attrs = {
+            "authenticationDate", "longTermAuthenticationRequestTokenUsed", "isFromNewLogin"
+        }
         ignored_attrs = 0
         attrs1 = set()
         for attr in attributes[0]:
             name = attr.tag[len("http://www.yale.edu/tp/cas")+2:]
-            if not name in ignore_attrs:
+            if name not in ignore_attrs:
                 attrs1.add((name, attr.text))
             else:
                 ignored_attrs += 1
@@ -164,7 +166,7 @@ class XmlContent(object):
         attrs2 = set()
         for attr in attributes:
             name = attr.attrib['name']
-            if not name in ignore_attrs:
+            if name not in ignore_attrs:
                 attrs2.add((name, attr.attrib['value']))
         original = set()
         for key, value in original_attributes.items():
diff --git a/cas_server/tests/test_view.py b/cas_server/tests/test_view.py
index ebdb8bd..193a8d4 100644
--- a/cas_server/tests/test_view.py
+++ b/cas_server/tests/test_view.py
@@ -1907,7 +1907,9 @@ class SamlValidateTestCase(TestCase, BaseServicePattern, XmlContent):
             "//samla:AttributeStatement/samla:Attribute",
             namespaces={'samla': "urn:oasis:names:tc:SAML:1.0:assertion"}
         )
-        ignore_attrs = {"authenticationDate", "longTermAuthenticationRequestTokenUsed", "isFromNewLogin"} - set(original_attributes.keys())
+        ignore_attrs = {
+            "authenticationDate", "longTermAuthenticationRequestTokenUsed", "isFromNewLogin"
+        } - set(original_attributes.keys())
         attrs = set()
         for attr in attributes:
             if not attr.attrib['AttributeName'] in ignore_attrs: