Fixup 6020610: shorten the test

lcmd-epfl · Dec 4, 2023 · ffcf669 · ffcf669
1 parent 6020610
commit ffcf669
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_slatm.py b/tests/test_slatm.py
@@ -7,15 +7,15 @@
 def test_slatm_global():
     path = os.path.dirname(os.path.realpath(__file__))
     v0 = np.load(f'{path}/data/slatm/slatm_global.npy')
-    xyzs = [f for f in sorted(glob.glob(f"{path}/data/slatm/*.xyz"))]
+    xyzs = sorted(glob.glob(f"{path}/data/slatm/*.xyz"))
     v = get_slatm_for_dataset(xyzs, progress=False, global_repr=True)
     assert(np.linalg.norm(v-v0)<1e-10)
 
 
 def test_slatm_local():
     path = os.path.dirname(os.path.realpath(__file__))
     v0 = np.load(f'{path}/data/slatm/slatm_local.npy')
-    xyzs = [f for f in sorted(glob.glob(f"{path}/data/slatm/*.xyz"))]
+    xyzs = sorted(glob.glob(f"{path}/data/slatm/*.xyz"))
     v = get_slatm_for_dataset(xyzs, progress=False)
     assert(np.linalg.norm(v-v0)<1e-10)