fixup

Author: PierreMarchand20

tests/test_distributed_operator.py

@@ -96,32 +96,28 @@ def test_distributed_operator(
     Y_2 = generator.mat_mat(X)
     assert np.linalg.norm(Y_1 - Y_2) / np.linalg.norm(Y_2) < epsilon
 
-    # # Test sub matrix vector product
-    # test_offset = int(nb_cols / 10)
-    # test_size = int(nb_cols / 10)
-    # x[0:test_offset] = 0
-    # x[test_offset + test_size :] = 0
-    # x_perm = np.zeros(nb_cols)
-    # source_permutation = source_cluster.get_permutation()
-    # x_perm[source_permutation] = x
+    # Test sub matrix vector product
+    test_offset = int(nb_cols / 10)
+    test_size = int(nb_cols / 10)
+    x[0:test_offset] = 0
+    x[test_offset + test_size :] = 0
+    x_perm = np.zeros(nb_cols)
+    source_permutation = source_cluster.get_permutation()
+    x_perm[source_permutation] = x
 
-    # y_1 = distributed_operator.internal_sub_vector_product_global_to_local(
-    #     x[test_offset : test_offset + test_size], test_offset
-    # )
-    # y_2_perm = generator.mat_vec(x_perm)
-    # target_permutation = target_cluster.get_permutation()
-    # y_2 = y_2_perm[target_permutation]
-    # local_target_cluster = target_cluster.get_cluster_on_partition(
-    #     mpi4py.MPI.COMM_WORLD.Get_rank()
-    # )
-    # target_offset = local_target_cluster.get_offset()
-    # target_size = local_target_cluster.get_size()
-    # print(
-    #     np.linalg.norm(y_1 - y_2[target_offset : target_offset + target_size])
-    #     / np.linalg.norm(y_2)
-    # )
-    # assert (
-    #     np.linalg.norm(y_1 - y_2[target_offset : target_offset + target_size])
-    #     / np.linalg.norm(y_2)
-    #     < (1 + 0.2) * epsilon
-    # )
+    y_1 = distributed_operator.internal_sub_vector_product_global_to_local(
+        x[test_offset : test_offset + test_size], test_offset
+    )
+    y_2_perm = generator.mat_vec(x_perm)
+    target_permutation = target_cluster.get_permutation()
+    y_2 = y_2_perm[target_permutation]
+    local_target_cluster = target_cluster.get_cluster_on_partition(
+        mpi4py.MPI.COMM_WORLD.Get_rank()
+    )
+    target_offset = local_target_cluster.get_offset()
+    target_size = local_target_cluster.get_size()
+    assert (
+        np.linalg.norm(y_1 - y_2[target_offset : target_offset + target_size])
+        / np.linalg.norm(y_2)
+        < (1 + 10) * epsilon
+    )