reduce snapshot accuracy requirement

Author: AllanChain

tests/__snapshots__/adaptor_test.ambr

@@ -24,38 +24,38 @@
   DeviceArray(0.8603, dtype=float32)
 # ---
 # name: TestFermiNet.test_kinetic
-  Array(0.2784566, dtype=float32)
+  Array(0.278457, dtype=float32)
 # ---
 # name: TestFermiNet.test_potential
-  Array(-0.7828356, dtype=float32)
+  Array(-0.782836, dtype=float32)
 # ---
 # name: TestFermiNet.test_walk
-  Array([[[0.4166998, 0.2729937, 0.9946592],
-          [-0.1636449, 0.5025202, -0.2179334],
-          [0.5774042, 0.5321482, -0.83614],
-          [-0.6533463, 0.2915913, 1.5937456],
-          [0.479838, -0.0993455, -0.2716153],
-          [-0.4078843, -2.4394894, 0.041675],
-          [0.0554786, 0.2439971, -0.0937411],
-          [0.2253472, 0.4246373, -0.7441298],
-          [0.3037185, 1.1247488, -0.7110634],
-          [0.0041618, -0.7565254, -1.5015912]]], dtype=float32)
+  Array([[[0.4167, 0.272994, 0.994659],
+          [-0.163645, 0.50252, -0.217933],
+          [0.577404, 0.532148, -0.83614],
+          [-0.653346, 0.291591, 1.593746],
+          [0.479838, -0.099346, -0.271615],
+          [-0.407884, -2.439489, 0.041675],
+          [0.055479, 0.243997, -0.093741],
+          [0.225347, 0.424637, -0.74413],
+          [0.303718, 1.124749, -0.711063],
+          [0.004162, -0.756525, -1.501591]]], dtype=float32)
 # ---
 # name: TestFermiNet.test_wf
-  Array(0.1269548, dtype=float32)
+  Array(0.126955, dtype=float32)
 # ---
 # name: TestHydrogen.test_kinetic
-  Array(1.0260653, dtype=float32)
+  Array(1.026065, dtype=float32)
 # ---
 # name: TestHydrogen.test_potential
-  Array(-1.5260653, dtype=float32)
+  Array(-1.526065, dtype=float32)
 # ---
 # name: TestHydrogen.test_walk
-  Array([[[-0.0269817, -0.0194543, 0.308942],
-          [0.4157366, -1.0177534, 0.7061033],
-          [1.1319326, -1.5159024, -1.4163642],
-          [-0.5633768, 0.0473477, 1.8883846],
-          [0.062221, -0.3534039, -0.3654794]]], dtype=float32)
+  Array([[[-0.026982, -0.019454, 0.308942],
+          [0.415737, -1.017753, 0.706103],
+          [1.131933, -1.515902, -1.416364],
+          [-0.563377, 0.047348, 1.888385],
+          [0.062221, -0.353404, -0.365479]]], dtype=float32)
 # ---
 # name: TestHydrogen.test_wf
   Array(-0.65528, dtype=float32)

tests/__snapshots__/numerical_test.ambr

@@ -1,36 +1,36 @@
 # serializer version: 1
 # name: test_antithetic
-  Array([[-0.0008045, 0.0011302, -0.0004956]], dtype=float32)
+  Array([[-0.000805, 0.00113, -0.000496]], dtype=float32)
 # ---
 # name: test_antithetic_zb
-  Array([[-0.0003859, -0.0012329, -0.0017145]], dtype=float32)
+  Array([[-0.000386, -0.001233, -0.001715]], dtype=float32)
 # ---
 # name: test_bare
-  Array([[0.0891896, -0.0699865, 0.1169906]], dtype=float32)
+  Array([[0.08919, -0.069986, 0.116991]], dtype=float32)
 # ---
 # name: test_density
   Array([[[75.0, 815.0, 98.0],
           [84.0, 820.0, 98.0]]], dtype=float32)
 # ---
 # name: test_energy
   list([
-    Array(-0.5000338, dtype=float32),
-    Array(-1.0248146, dtype=float32),
-    Array(0.5247842, dtype=float32),
+    Array(-0.500034, dtype=float32),
+    Array(-1.024815, dtype=float32),
+    Array(0.524784, dtype=float32),
   ])
 # ---
 # name: test_swct
   tuple(
     Array([[0.0, 0.0, 0.0]], dtype=float32),
-    Array([[9.41e-05, -6.9e-05, 0.0001135]], dtype=float32),
+    Array([[9.4e-05, -6.9e-05, 0.000114]], dtype=float32),
   )
 # ---
 # name: test_wf_change
-  Array(0.9999958, dtype=float32)
+  Array(0.999996, dtype=float32)
 # ---
 # name: test_zvzb
   tuple(
-    Array([[3.4e-06, 8.9e-06, 1.2e-06]], dtype=float32),
+    Array([[3e-06, 9e-06, 1e-06]], dtype=float32),
     Array([[0.0, 0.0, 0.0]], dtype=float32),
   )
 # ---

tests/conftest.py

@@ -20,7 +20,7 @@
 def consistent_jax_print():
     def float_round(x):
         # Simply rounding will have "0.0" vs "-0.0" issue
-        return "0.0" if abs(x) < 1e-7 else str(round(x, 7))
+        return "0.0" if abs(x) < 1e-6 else str(round(x, 6))
 
     jnp.set_printoptions(
         suppress=True,