Fix tests and formatting

Author: edan-bainglass

src/aiida_pseudo/groups/family/pseudo.py

@@ -347,11 +347,7 @@ def get_pseudos(
 
         if structure is not None:
             return {
-                kind.name: self.get_pseudo(kind.symbol, none_if_missing=none_if_missing)
-                for kind in structure.kinds
+                kind.name: self.get_pseudo(kind.symbol, none_if_missing=none_if_missing) for kind in structure.kinds
             }
 
-        return {
-            element: self.get_pseudo(element, none_if_missing=none_if_missing)
-            for element in elements
-        }
+        return {element: self.get_pseudo(element, none_if_missing=none_if_missing) for element in elements}

tests/groups/family/test_pseudo.py

@@ -431,7 +431,8 @@ def test_get_pseudos_raise(get_pseudo_family, generate_structure):
     with pytest.raises(ValueError, match=r'family `.*` does not contain pseudo for element `.*`'):
         family.get_pseudos(structure=structure)
 
-    assert family.get_pseudos(structure=structure, none_if_missing=True)['He'] is None
+    assert family.get_pseudos(structure=structure, none_if_missing=True)['Ne'] is None
+
 
 @pytest.mark.usefixtures('aiida_profile_clean')
 def test_get_pseudos_list(get_pseudo_family):