From 8732fa10bb258bbda189a24a9453c28431b98d2e Mon Sep 17 00:00:00 2001 From: Kaushik Kulkarni <kaushikcfd@gmail.com> Date: Sat, 14 Aug 2021 16:34:47 -0500 Subject: [PATCH] defines DictOfNamedArrays.__eq__ --- pytato/array.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/pytato/array.py b/pytato/array.py index 25ffc52..8fe6ea4 100644 --- a/pytato/array.py +++ b/pytato/array.py @@ -815,6 +815,13 @@ class DictOfNamedArrays(AbstractResultWithNamedArrays): def __iter__(self) -> Iterator[str]: return iter(self._data) + def __eq__(self, other: Any) -> bool: + if self is other: + return True + + return (isinstance(other, DictOfNamedArrays) + and self._data == other._data) + # }}} -- GitLab