Improved test

luigibonati · Apr 4, 2024 · d7d8c8b · d7d8c8b
1 parent 52d8061
commit d7d8c8b
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/mlcolvar/core/transform/normalization.py b/mlcolvar/core/transform/normalization.py
@@ -206,19 +206,17 @@ def test_normalization():
     norm = Normalization(in_features, mean=stats["mean"], range=stats["std"])
 
     y = norm(X)
-    # print(X.mean(0),y.mean(0))
-    # print(X.std(0),y.std(0))
 
     # test inverse
     z = norm.inverse(y)
-    # print(X.mean(0),z.mean(0))
-    # print(X.std(0),z.std(0))
+    assert(torch.allclose(X.mean(0), z.mean(0)))
+    assert(torch.allclose(X.std(0) , z.std(0)))
 
     # test inverse class
     inverse = Inverse(norm)
     q = inverse(y)
-    # print(X.mean(0),q.mean(0))
-    # print(X.std(0),q.std(0))
+    assert(torch.allclose(X.mean(0), q.mean(0)))
+    assert(torch.allclose(X.std(0), q.std(0)))
     norm = Normalization(
         in_features, mean=stats["mean"], range=stats["std"], mode="min_max"
     )