Add __eq__ to SolidityImportPlaceHolder

crytic · Dec 17, 2020 · d1eee22 · d1eee22
1 parent 72738bc
commit d1eee22
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/slither/core/declarations/solidity_variables.py b/slither/core/declarations/solidity_variables.py
@@ -204,6 +204,13 @@ def _check_name(self, name: str):
     def type(self) -> ElementaryType:
         return ElementaryType("string")
 
+    def __eq__(self, other):
+        return (
+            self.__class__ == other.__class__
+            and self.name == other.name
+            and self._import_directive.filename == self._import_directive.filename
+        )
+
     @property
     def import_directive(self) -> "Import":
         return self._import_directive