Fix != operations on lazy objects.

django · Apr 19, 2013 · 714161c · 714161c
1 parent 59d127e
commit 714161c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/django/utils/functional.py b/django/utils/functional.py
@@ -346,6 +346,7 @@ def __repr__(self):
     # care about this (especially in equality tests)
     __class__ = property(new_method_proxy(operator.attrgetter("__class__")))
     __eq__ = new_method_proxy(operator.eq)
+    __ne__ = new_method_proxy(operator.ne)
     __hash__ = new_method_proxy(hash)
     __bool__ = new_method_proxy(bool)       # Python 3
     __nonzero__ = __bool__                  # Python 2

diff --git a/tests/utils_tests/test_simplelazyobject.py b/tests/utils_tests/test_simplelazyobject.py
@@ -152,3 +152,12 @@ def trace_func(frame, event, arg):
             SimpleLazyObject(None)
         finally:
             sys.settrace(old_trace_func)
+
+    def test_not_equal(self):
+        lazy1 = SimpleLazyObject(lambda: 2)
+        lazy2 = SimpleLazyObject(lambda: 2)
+        lazy3 = SimpleLazyObject(lambda: 3)
+        self.assertEqual(lazy1, lazy2)
+        self.assertNotEqual(lazy1, lazy3)
+        self.assertTrue(lazy1 != lazy3)
+        self.assertFalse(lazy1 != lazy2)