one more test and general simplication

Author: guglielmopadula

tests/test_bffd.py

@@ -14,8 +14,6 @@ def test_nothing_happens(self):
         A = np.random.rand(3, original_mesh_points.reshape(-1).shape[0])
         b = cffd.fun(original_mesh_points)
         cffd.fixval = b
-        cffd.indices = np.arange(np.prod(cffd.n_control_points) * 3).tolist()
-        cffd.M = np.eye(len(cffd.indices))
         new_mesh_points = cffd(original_mesh_points)
         assert np.array_equal(original_mesh_points, new_mesh_points)
 
@@ -28,8 +26,6 @@ def test_constraint(self):
             "tests/test_datasets/meshpoints_sphere_orig.npy")
         b = cffd.fun(original_mesh_points)
         cffd.fixval = b
-        cffd.indices = np.arange(np.prod(cffd.n_control_points) * 3).tolist()
-        cffd.M = np.eye(len(cffd.indices))
         new_mesh_points = cffd(original_mesh_points)
         assert np.isclose(np.linalg.norm(cffd.fun(new_mesh_points) - b),
                           np.array([0.0]))

tests/test_cffd.py

@@ -6,6 +6,20 @@
 
 
 class TestCFFD(TestCase):
+    def test_nothing_happens_0(self):
+        np.random.seed(0)
+        cffd = CFFD()
+        original_mesh_points = np.load(
+            "tests/test_datasets/meshpoints_sphere_orig.npy")
+        A = np.random.rand(3, original_mesh_points.reshape(-1).shape[0])
+
+        def fun(x):
+            x = x.reshape(-1)
+            return A @ x
+        new_mesh_points = cffd(original_mesh_points)
+        assert np.array_equal(original_mesh_points, new_mesh_points)
+
+
     def test_nothing_happens(self):
         np.random.seed(0)
         cffd = CFFD()