223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

added __eq__ and test

This commit is contained in:
Philip Eisenlohr 2021-11-23 17:46:38 -05:00
parent 3e1a6dcab7
commit 3bf10127b8
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
python/damask/_colormap.py
View File

@ -42,6 +42,11 @@ class Colormap(mpl.colors.ListedColormap):
""" """
def __eq__(self, other) -> bool:
"""Test equality of colormaps."""
return len(self.colors) == len(other.colors) \
and bool(np.all(self.colors == other.colors))
def __add__(self, other: "Colormap") -> "Colormap": def __add__(self, other: "Colormap") -> "Colormap":
"""Concatenate.""" """Concatenate."""
return Colormap(np.vstack((self.colors,other.colors)), return Colormap(np.vstack((self.colors,other.colors)),