From 0a95b13f36932bafb8bbb0d7c32b0895f50f562e Mon Sep 17 00:00:00 2001
From: pierregm <pierregm@localhost>
Date: Tue, 29 Jun 2010 17:57:15 +0000
Subject: Fixed __eq__/__ne__ for scalars

---
 numpy/ma/tests/test_core.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'numpy/ma/tests')

diff --git a/numpy/ma/tests/test_core.py b/numpy/ma/tests/test_core.py
index f95d621db..908d7adc6 100644
--- a/numpy/ma/tests/test_core.py
+++ b/numpy/ma/tests/test_core.py
@@ -1149,6 +1149,21 @@ class TestMaskedArrayArithmetic(TestCase):
         assert_equal(test.mask, [False, False])
 
 
+    def test_eq_w_None(self):
+        a = array([1, 2], mask=False)
+        assert_equal(a == None, False)
+        assert_equal(a != None, True)
+        a = masked
+        assert_equal(a == None, masked)
+
+    def test_eq_w_scalar(self):
+        a = array(1)
+        assert_equal(a == 1, True)
+        assert_equal(a == 0, False)
+        assert_equal(a != 1, False)
+        assert_equal(a != 0, True)
+
+
     def test_numpyarithmetics(self):
         "Check that the mask is not back-propagated when using numpy functions"
         a = masked_array([-1, 0, 1, 2, 3], mask=[0, 0, 0, 0, 1])
-- 
cgit v1.2.1