diff --git a/tests/test_abacus_deltaspin.py b/tests/test_abacus_deltaspin.py index d57b3cf4..f26369dd 100644 --- a/tests/test_abacus_deltaspin.py +++ b/tests/test_abacus_deltaspin.py @@ -1,29 +1,23 @@ from __future__ import annotations -import os -import shutil import unittest -import numpy as np from context import dpdata -from dpdata.unit import LengthConversion -class TestDeltaSpin(): - """Make a test for the DeltaSpin module in ABACUS - - """ +class TestDeltaSpin: + """Make a test for the DeltaSpin module in ABACUS""" def OUTABACUS_to_system(): - data = dpdata.LabeledSystem('./abacus.scf.deltaspin', fmt = 'abacus/scf') + data = dpdata.LabeledSystem("./abacus.scf.deltaspin", fmt="abacus/scf") print(data) - + def system_to_npy(): - data = dpdata.LabeledSystem('./abacus.scf.deltaspin', fmt = 'abacus/scf').to_deepmd_npy('./abacus.scf.deltaspin/') + data = dpdata.LabeledSystem( + "./abacus.scf.deltaspin", fmt="abacus/scf" + ).to_deepmd_npy("./abacus.scf.deltaspin/") print(data) - - if __name__ == "__main__": - unittest.main() \ No newline at end of file + unittest.main()