4
4
from context import dpdata
5
5
from comp_sys import CompLabeledSys
6
6
7
- class TestQuipGapxyz (unittest .TestCase , CompLabeledSys ):
7
+ class TestQuipGapxyz1 (unittest .TestCase , CompLabeledSys ):
8
8
def setUp (self ) :
9
9
self .multi_systems = dpdata .MultiSystems .from_file ('xyz/xyz_unittest.xyz' ,'quip/gap/xyz' )
10
10
self .system_1 = self .multi_systems .systems ['B1C9' ]
@@ -17,7 +17,7 @@ def setUp (self) :
17
17
class TestQuipGapxyz2 (unittest .TestCase , CompLabeledSys ):
18
18
def setUp (self ) :
19
19
self .system_temp0 = dpdata .MultiSystems .from_file (file_name = 'xyz/xyz_unittest.xyz' , fmt = 'quip/gap/xyz' )
20
- self .system_1 = self .system_temp0 .systems ['B5C7' ]
20
+ self .system_1 = self .system_temp0 .systems ['B5C7' ] # .sort_atom_types()
21
21
self .system_temp1 = dpdata .LabeledSystem ('xyz/B1C9' , fmt = 'deepmd' )
22
22
self .system_temp2 = dpdata .LabeledSystem ('xyz/B5C7' , fmt = 'deepmd' )
23
23
self .system_temp3 = dpdata .MultiSystems (self .system_temp2 , self .system_temp1 )
@@ -27,5 +27,54 @@ def setUp (self) :
27
27
self .f_places = 6
28
28
self .v_places = 4
29
29
30
+ class TestQuipGapxyzsort1 (unittest .TestCase , CompLabeledSys ):
31
+ def setUp (self ) :
32
+ self .multi_systems_1 = dpdata .MultiSystems .from_file ('xyz/xyz_unittest.sort.xyz' ,'quip/gap/xyz' )
33
+ self .system_1 = self .multi_systems_1 .systems ['B5C7' ]
34
+ self .system_1 .sort_atom_types ()
35
+ self .multi_systems_2 = dpdata .MultiSystems .from_file ('xyz/xyz_unittest.xyz' ,'quip/gap/xyz' )
36
+ self .system_2 = self .multi_systems_2 .systems ['B5C7' ]
37
+ self .places = 6
38
+ self .e_places = 6
39
+ self .f_places = 6
40
+ self .v_places = 4
41
+
42
+ class TestQuipGapxyzsort2 (unittest .TestCase , CompLabeledSys ):
43
+ def setUp (self ) :
44
+ self .multi_systems_1 = dpdata .MultiSystems .from_file ('xyz/xyz_unittest.sort.xyz' ,'quip/gap/xyz' )
45
+ self .system_1 = self .multi_systems_1 .systems ['B1C9' ]
46
+ self .system_1 .sort_atom_types ()
47
+ self .multi_systems_2 = dpdata .MultiSystems .from_file ('xyz/xyz_unittest.xyz' ,'quip/gap/xyz' )
48
+ self .system_2 = self .multi_systems_2 .systems ['B1C9' ]
49
+ self .places = 6
50
+ self .e_places = 6
51
+ self .f_places = 6
52
+ self .v_places = 4
53
+
54
+ class TestQuipGapxyzfield (unittest .TestCase , CompLabeledSys ):
55
+ def setUp (self ) :
56
+ self .multi_systems_1 = dpdata .MultiSystems .from_file ('xyz/xyz_unittest.field.xyz' ,'quip/gap/xyz' )
57
+ self .system_1 = self .multi_systems_1 .systems ['B1C9' ]
58
+ self .system_1 .sort_atom_types ()
59
+ self .multi_systems_2 = dpdata .MultiSystems .from_file ('xyz/xyz_unittest.xyz' ,'quip/gap/xyz' )
60
+ self .system_2 = self .multi_systems_2 .systems ['B1C9' ]
61
+ self .places = 6
62
+ self .e_places = 6
63
+ self .f_places = 6
64
+ self .v_places = 4
65
+
66
+ class TestQuipGapxyzfield2 (unittest .TestCase , CompLabeledSys ):
67
+ def setUp (self ) :
68
+ self .multi_systems_1 = dpdata .MultiSystems .from_file ('xyz/xyz_unittest.field.xyz' ,'quip/gap/xyz' )
69
+ self .system_1 = self .multi_systems_1 .systems ['B5C7' ]
70
+ self .system_1 .sort_atom_types ()
71
+ self .multi_systems_2 = dpdata .MultiSystems .from_file ('xyz/xyz_unittest.xyz' ,'quip/gap/xyz' )
72
+ self .system_2 = self .multi_systems_2 .systems ['B5C7' ]
73
+ self .places = 6
74
+ self .e_places = 6
75
+ self .f_places = 6
76
+ self .v_places = 4
77
+
78
+
30
79
if __name__ == '__main__' :
31
80
unittest .main ()
0 commit comments