@@ -162,11 +162,19 @@ def test_assignUniqueLabels(self):
162162 self .assertEqual ("C2" , self .stru [1 ].label )
163163 return
164164
165+ def test_assign_unique_labels (self ):
166+ """Check Structure.assign_unique_labels()"""
167+ self .assertEqual ("" , "" .join ([a .label for a in self .stru ]))
168+ self .stru .assign_unique_labels ()
169+ self .assertEqual ("C1" , self .stru [0 ].label )
170+ self .assertEqual ("C2" , self .stru [1 ].label )
171+ return
172+
165173 def test_distance (self ):
166174 """Check Structure.distance()"""
167175 from math import sqrt
168176
169- self .stru .assignUniqueLabels ()
177+ self .stru .assign_unique_labels ()
170178 self .assertRaises (IndexError , self .stru .distance , 333 , "C1" )
171179 self .assertRaises (IndexError , self .stru .distance , "C" , "C1" )
172180 self .assertAlmostEqual (sqrt (2.0 ), self .stru .distance (0 , 1 ), self .places )
@@ -177,7 +185,7 @@ def test_distance(self):
177185 def test_angle (self ):
178186 """Check Structure.angle()"""
179187 cdse = Structure (filename = self .cdsefile )
180- cdse .assignUniqueLabels ()
188+ cdse .assign_unique_labels ()
181189 self .assertEqual (109 , round (cdse .angle (0 , 2 , 1 )))
182190 self .assertEqual (109 , round (cdse .angle ("Cd1" , "Se1" , "Cd2" )))
183191 return
@@ -281,7 +289,7 @@ def test___getitem__(self):
281289 cdse013 .pop (2 )
282290 self .assertEqual (cdse013 , cdse [:2 , 3 ].tolist ())
283291 self .assertRaises (IndexError , cdse .__getitem__ , "Cd1" )
284- cdse .assignUniqueLabels ()
292+ cdse .assign_unique_labels ()
285293 self .assertTrue (cdse [0 ] is cdse ["Cd1" ])
286294 cdse [0 ].label = "Hohenzollern"
287295 self .assertRaises (IndexError , cdse .__getitem__ , "Cd1" )
@@ -520,7 +528,7 @@ def test_label(self):
520528 """Check Structure.label."""
521529 cdse = Structure (filename = self .cdsefile )
522530 self .assertEqual (4 * ["" ], cdse .label .tolist ())
523- cdse .assignUniqueLabels ()
531+ cdse .assign_unique_labels ()
524532 self .assertEqual ("Cd1 Cd2 Se1 Se2" .split (), cdse .label .tolist ())
525533 cdse .label = cdse .label .lower ()
526534 self .assertEqual ("cd1 cd2 se1 se2" .split (), cdse .label .tolist ())
0 commit comments