Add NdRange.__eq__ method to make object equality checks possible.

Author: Diptorup Deb

numba_dpex/core/kernel_interface/indexers.py

@@ -219,6 +219,12 @@ def __repr__(self):
         """
         return self.__str__()
 
+    def __eq__(self, other):
+        return (
+            self.global_range == other.global_range
+            and self.local_range == other.local_range
+        )
+
 
 @intrinsic
 def _intrin_range_alloc(typingctx, ty_dim0, ty_dim1, ty_dim2, ty_range):

numba_dpex/tests/core/types/range_types/test_constructor_overloads.py

@@ -18,10 +18,7 @@ def _tester(r):
     r_expected = Range(*r)
     r_out = _tester(r)
 
-    assert r_out.ndim == r_expected.ndim
-    assert r_out.dim0 == r_expected.dim0
-    assert r_out.dim1 == r_expected.dim1
-    assert r_out.dim2 == r_expected.dim2
+    assert r_out == r_expected
 
 
 @pytest.mark.parametrize("r", ranges)
@@ -35,11 +32,4 @@ def _tester(r):
     r_expected = NdRange(gr, lr)
     r_out = _tester(r)
 
-    assert r_out.global_range.ndim == r_expected.global_range.ndim
-    assert r_out.local_range.ndim == r_expected.local_range.ndim
-    assert r_out.global_range.dim0 == r_expected.global_range.dim0
-    assert r_out.global_range.dim1 == r_expected.global_range.dim1
-    assert r_out.global_range.dim2 == r_expected.global_range.dim2
-    assert r_out.local_range.dim0 == r_expected.local_range.dim0
-    assert r_out.local_range.dim1 == r_expected.local_range.dim1
-    assert r_out.local_range.dim2 == r_expected.local_range.dim2
+    assert r_out == r_expected

numba_dpex/tests/core/types/range_types/test_unbox_box.py

@@ -18,10 +18,7 @@ def _tester(r):
     r_in = Range(*r)
     r_out = _tester(r_in)
 
-    assert r_out.ndim == r_in.ndim
-    assert r_out.dim0 == r_in.dim0
-    assert r_out.dim1 == r_in.dim1
-    assert r_out.dim2 == r_in.dim2
+    assert r_out == r_in
 
 
 @pytest.mark.parametrize("r", ranges)
@@ -34,11 +31,4 @@ def _tester(r):
     r_in = NdRange(gr, lr)
     r_out = _tester(r_in)
 
-    assert r_out.global_range.ndim == r_in.global_range.ndim
-    assert r_out.local_range.ndim == r_in.local_range.ndim
-    assert r_out.global_range.dim0 == r_in.global_range.dim0
-    assert r_out.global_range.dim1 == r_in.global_range.dim1
-    assert r_out.global_range.dim2 == r_in.global_range.dim2
-    assert r_out.local_range.dim0 == r_in.local_range.dim0
-    assert r_out.local_range.dim1 == r_in.local_range.dim1
-    assert r_out.local_range.dim2 == r_in.local_range.dim2
+    assert r_out == r_in