adding setget tests, removing debug prints

Author: fschlimb

src/Deferred.cpp

@@ -42,7 +42,6 @@ void process_promises()
     while(true) {
         Deferred::ptr_type d;
         _deferred.pop(d);
-        std::cerr << "Executing something" << std::endl;
         if(d) d->run();
         else break;
         d.reset();

src/MPIMediator.cpp

@@ -145,7 +145,6 @@ void MPIMediator::listen()
         Deserializer ser{buff.begin(), cnt};
         int tag;
         ser.value<sizeof(tag)>(tag);
-        std::cerr << "Recvd " << cnt << " bytes from " << requester << " tag " << tag << std::endl;
 
         switch(tag) {
         case DEFER_TAG: {
@@ -170,7 +169,6 @@ void MPIMediator::listen()
             MPI_Wait(&request_out, MPI_STATUS_IGNORE);
             ptr->bufferize(slice, rbuff);
             if(slice.size() * ptr->item_size() != rbuff.size()) throw(std::runtime_error("Got unexpected buffer size."));
-            std::cerr << "Sending " << rbuff.size() << " bytes to " << requester << " tag PUSH_TAG" << std::endl;
             MPI_Isend(rbuff.data(), rbuff.size(), MPI_CHAR, requester, PUSH_TAG, comm, &request_out);
             break;
         }

test/conftest.py

@@ -1,11 +1,12 @@
 import os
 import ddptensor as dt
 
-ddpt_cw = bool(int(os.environ.get('DDPT_CW')))
+ddpt_cw = os.environ.get('DDPT_CW')
+ddpt_cw = False if ddpt_cw is None else bool(int(ddpt_cw))
 
 def pytest_configure(config):
-    print(f"DDPT_CW={bool(ddpt_cw)}")
     dt.init(ddpt_cw)
 
 def pytest_unconfigure(config):
     dt.fini()
+    #pytest.exit("", returncode=0)

test/test_set.py

@@ -0,0 +1,31 @@
+import ddptensor as dt
+
+class TestSetGet:
+    def test_setitem1(self):
+        a = dt.ones([16, 16], dt.float64)
+        a[0:16:2, 0:16:2] = dt.zeros([8,8], dt.float64)
+        for i in range(16):
+            for j in range(16):
+                assert float(a[i:i+1, j:j+1]) == (1 if i % 2 or j % 2 else 0)
+
+    def test_setitem2(self):
+        a = dt.ones([16, 16], dt.float64)
+        b = dt.zeros([16, 16], dt.float64)
+        a[1:8,0:6] = b[0:7, 0:6]
+        for i in range(16):
+            for j in range(16):
+                if i >= 1 and i < 8 and j < 6:
+                    assert float(a[i:i+1, j:j+1]) == 0
+                else:
+                    assert float(a[i:i+1, j:j+1]) == 1
+
+    def test_setitem3(self):
+        a = dt.ones([16, 16], dt.float64)
+        b = dt.zeros([16, 16], dt.float64)
+        a[7:16:3, 4:10:2] = b[4:7, 10:16:2]
+        for i in range(16):
+            for j in range(16):
+                if i >= 7 and i < 16 and ((i-7) % 3 == 0) and j >= 4 and j < 10 and ((j-4) % 2 == 0):
+                    assert float(a[i:i+1, j:j+1]) == 0
+                else:
+                    assert float(a[i:i+1, j:j+1]) == 1