TEST: Add test for array comparison

Author: ubaidsk

integration_tests/CMakeLists.txt

@@ -368,6 +368,7 @@ RUN(NAME array_size_02            LABELS cpython llvm c)
 RUN(NAME array_01            LABELS cpython llvm wasm c)
 RUN(NAME array_02            LABELS cpython wasm c)
 RUN(NAME array_03            LABELS cpython llvm c)
+RUN(NAME array_04            LABELS cpython llvm c)
 RUN(NAME bindc_01            LABELS cpython llvm c)
 RUN(NAME bindc_02            LABELS cpython llvm c)
 RUN(NAME bindc_04            LABELS llvm c NOFAST)

integration_tests/array_04.py

@@ -0,0 +1,19 @@
+from lpython import i32, Const
+from numpy import empty, int32
+
+def main0():
+    n: Const[i32] = 1
+    x: i32[n, n] = empty([n, n], dtype=int32)
+    y: i32[n, n] = empty([n, n], dtype=int32)
+
+    x[0, 0] = -10
+    y[0, 0] = -10
+
+    print(x[0, 0], y[0, 0])
+    assert x == y
+
+    y[0, 0] = 10
+    print(x[0, 0], y[0, 0])
+    assert x != y
+
+main0()

tests/errors/arrays_02.py

@@ -0,0 +1,28 @@
+from lpython import (i8, i32, dataclass)
+from numpy import (empty, int8)
+
+@dataclass
+class Foo:
+    a : i8[4] = empty(4, dtype=int8)
+    dim : i32 = 4
+
+def trinary_majority(x : Foo, y : Foo, z : Foo) -> Foo:
+    foo : Foo = Foo()
+    i : i32
+    for i in range(foo.dim):
+        foo.a[i] = (x.a[i] & y.a[i]) | (y.a[i] & z.a[i]) | (z.a[i] & x.a[i])
+    return foo
+
+
+t1 : Foo = Foo()
+t1.a = empty(4, dtype=int8)
+
+t2 : Foo = Foo()
+t2.a = empty(4, dtype=int8)
+
+t3 : Foo = Foo()
+t3.a = empty(4, dtype=int8)
+
+r1 : Foo = trinary_majority(t1, t2, t3)
+
+assert r1.a == t1.a

tests/tests.toml

@@ -650,6 +650,10 @@ asr = true
 filename = "errors/arrays_01.py"
 asr = true
 
+[[test]]
+filename = "errors/arrays_02.py"
+asr = true
+
 [[test]]
 filename = "errors/structs_02.py"
 asr = true