Add missing Trie.__bool__ method

mina86 · May 5, 2018 · 19ae72a · 19ae72a
1 parent f2df4da
commit 19ae72a
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/pygtrie.py b/pygtrie.py
@@ -178,11 +178,8 @@ def __eq__(self, other):
     def __ne__(self, other):
         return not self.__eq__(other)
 
-    def __bool__(self):
-        return bool(self.value is not _SENTINEL or self.children)
-
-    __nonzero__ = __bool__
-
+    __bool__ = __nonzero__ = lambda self: bool(
+        self.value is not _SENTINEL or self.children)
     __hash__ = None
 
     def __getstate__(self):
@@ -535,8 +532,7 @@ def __len__(self):
         """
         return sum(1 for _ in self.itervalues())
 
-    def __nonzero__(self):
-        return bool(self._root)
+    __bool__ = __nonzero__ = lambda self: bool(self._root)
 
     HAS_VALUE = 1
     HAS_SUBTRIE = 2