diff --git a/python/block_structure.py b/python/block_structure.py index 5bb5f823..2dcd7c81 100644 --- a/python/block_structure.py +++ b/python/block_structure.py @@ -376,10 +376,10 @@ class BlockStructure(object): if not compare(x,y): return False return True - elif isinstance(one,int): - return one==two - elif isinstance(one,str): + elif isinstance(one,(int,bool, str, np.bool_)): return one==two + elif isinstance(one,np.ndarray): + return np.all(one==two) elif isinstance(one,dict): if set(one.keys()) != set(two.keys()): return False @@ -391,7 +391,8 @@ class BlockStructure(object): return False for prop in [ "gf_struct_sumk", "gf_struct_solver", - "solver_to_sumk", "sumk_to_solver", "solver_to_sumk_block"]: + "solver_to_sumk", "sumk_to_solver", "solver_to_sumk_block", + "deg_shells"]: if not compare(getattr(self,prop),getattr(other,prop)): return False return True diff --git a/test/blockstructure.ref.h5 b/test/blockstructure.ref.h5 index b290411c..c9eb4230 100644 Binary files a/test/blockstructure.ref.h5 and b/test/blockstructure.ref.h5 differ