Skip to content
Snippets Groups Projects
Commit a1efd95f authored by Jens J. Mortensen's avatar Jens J. Mortensen
Browse files

Implement 'in' operator

parent 31dbd14d
Branches
Tags
No related merge requests found
......@@ -32,6 +32,9 @@ class Reader:
self.shapes[name].append(n)
self.dims[dim.attrib['name']] = n
def __contains__(self, name):
return name in self.parameters or name in self.shapes
def __getattr__(self, name):
if name in self.parameters:
return self.parameters[name]
......@@ -59,3 +62,4 @@ class Reader:
if __name__ == '__main__':
r = Reader('H2.gpw')
print(r.Eigenvalues, r.FermiLevel, r.UnitCell)
print('UnitCell' in r, 'sdgsdgsdfg' in r)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment