@@ -60,9 +60,9 @@ def testContainment(self):
60
60
scope."""
61
61
c = makeC60 ()
62
62
m = self .c .GetScatterer ("c60" )
63
- self .assertEquals ("c60" , m .GetName ())
63
+ self .assertEqual ("c60" , m .GetName ())
64
64
del c
65
- self .assertEquals ("c60" , m .GetName ())
65
+ self .assertEqual ("c60" , m .GetName ())
66
66
return
67
67
68
68
def testAddPar (self ):
@@ -72,11 +72,11 @@ def testAddPar(self):
72
72
rpt = RefParType ("test" )
73
73
par = RefinablePar ("testpar" , 3 , 0 , 10 , rpt )
74
74
m .AddPar (par )
75
- self .assertAlmostEquals (3 , par .GetValue ())
75
+ self .assertAlmostEqual (3 , par .GetValue ())
76
76
del m
77
- self .assertAlmostEquals (3 , par .GetValue ())
77
+ self .assertAlmostEqual (3 , par .GetValue ())
78
78
del c
79
- self .assertAlmostEquals (3 , par .GetValue ())
79
+ self .assertAlmostEqual (3 , par .GetValue ())
80
80
return
81
81
82
82
def testAtoms (self ):
@@ -101,7 +101,7 @@ def testAtoms(self):
101
101
102
102
# Make sure the atom is still valid. We don't want RemoveAtom deleting
103
103
# the memory for an object we still have access to.
104
- self .assertEquals (a .X , x )
104
+ self .assertEqual (a .X , x )
105
105
106
106
# Check to see if a is in our list
107
107
for i in xrange (59 ):
@@ -121,9 +121,9 @@ def testAtoms(self):
121
121
self .m .RemoveAtom (a )
122
122
123
123
atoms = self .m .GetAtomList ()
124
- self .assertEquals (0 , len (atoms ))
124
+ self .assertEqual (0 , len (atoms ))
125
125
126
- self .assertEquals (0 , self .m .GetNbAtoms ())
126
+ self .assertEqual (0 , self .m .GetNbAtoms ())
127
127
128
128
return
129
129
@@ -186,7 +186,7 @@ def testBonds(self):
186
186
187
187
# Remove the bond
188
188
bonds = self .m .GetBondList ()
189
- self .assertEquals (1 , len (bonds ))
189
+ self .assertEqual (1 , len (bonds ))
190
190
self .m .RemoveBond (bonds [0 ])
191
191
# is the bond still in existance?
192
192
self .assertEqual (name , bond8 .GetName ())
@@ -201,7 +201,7 @@ def testBonds(self):
201
201
a = bond10 .GetAtom1 ()
202
202
# Try to remove that atom
203
203
self .m .RemoveAtom (a )
204
- self .assertEquals (0 , self .m .GetNbBonds ())
204
+ self .assertEqual (0 , self .m .GetNbBonds ())
205
205
206
206
return
207
207
@@ -259,7 +259,7 @@ def testBondAngles(self):
259
259
260
260
# Remove the bond angle
261
261
angles = self .m .GetBondAngleList ()
262
- self .assertEquals (1 , len (angles ))
262
+ self .assertEqual (1 , len (angles ))
263
263
self .m .RemoveBondAngle (angles [0 ])
264
264
# is the object still in existance?
265
265
self .assertEqual (name , ba8 .GetName ())
@@ -274,7 +274,7 @@ def testBondAngles(self):
274
274
a = ba10 .GetAtom1 ()
275
275
# Try to remove that atom
276
276
self .m .RemoveAtom (a )
277
- self .assertEquals (0 , self .m .GetNbBondAngles ())
277
+ self .assertEqual (0 , self .m .GetNbBondAngles ())
278
278
return
279
279
280
280
def testDihedralAngles (self ):
@@ -328,7 +328,7 @@ def testDihedralAngles(self):
328
328
329
329
# Remove the dihedral angle
330
330
angles = self .m .GetDihedralAngleList ()
331
- self .assertEquals (1 , len (angles ))
331
+ self .assertEqual (1 , len (angles ))
332
332
self .m .RemoveDihedralAngle (angles [0 ])
333
333
# is the object still in existance?
334
334
self .assertEqual (name , da8 .GetName ())
@@ -343,29 +343,29 @@ def testDihedralAngles(self):
343
343
a = da10 .GetAtom1 ()
344
344
# Try to remove that atom
345
345
self .m .RemoveAtom (a )
346
- self .assertEquals (0 , self .m .GetNbDihedralAngles ())
346
+ self .assertEqual (0 , self .m .GetNbDihedralAngles ())
347
347
348
348
return
349
349
350
350
def testRigidGroup (self ):
351
351
"""Test adding and manipulating a rigid group."""
352
352
# A rigid group has the interface of a set
353
- self .assertEquals (0 , len (self .m .GetRigidGroupList ()))
354
- self .assertEquals (0 , self .m .GetNbRigidGroups ())
353
+ self .assertEqual (0 , len (self .m .GetRigidGroupList ()))
354
+ self .assertEqual (0 , self .m .GetNbRigidGroups ())
355
355
rg = self .m .AddRigidGroup (self .m .GetAtomList ())
356
- self .assertEquals (1 , self .m .GetNbRigidGroups ())
356
+ self .assertEqual (1 , self .m .GetNbRigidGroups ())
357
357
rgl = self .m .GetRigidGroupList ()
358
- self .assertEquals (1 , len (rgl ))
359
- self .assertEquals (60 , len (rgl [0 ]))
358
+ self .assertEqual (1 , len (rgl ))
359
+ self .assertEqual (60 , len (rgl [0 ]))
360
360
361
361
# We would like to check to see if the atoms are the same, but the
362
362
# rigid group is a set, not a list.
363
363
364
364
# Test to see if we can remove the list.
365
365
self .m .RemoveRigidGroup (rg )
366
- self .assertEquals (0 , self .m .GetNbRigidGroups ())
366
+ self .assertEqual (0 , self .m .GetNbRigidGroups ())
367
367
rgl = self .m .GetRigidGroupList ()
368
- self .assertEquals (0 , len (rgl ))
368
+ self .assertEqual (0 , len (rgl ))
369
369
370
370
return
371
371
@@ -384,15 +384,15 @@ def testManipulation(self):
384
384
385
385
self .m .TranslateAtomGroup (self .m .GetAtomList (), 0 , 0 , 0.5 )
386
386
387
- self .assertAlmostEquals (x , a0 .X )
388
- self .assertAlmostEquals (y , a0 .Y )
389
- self .assertAlmostEquals (z + 0.5 , a0 .Z )
387
+ self .assertAlmostEqual (x , a0 .X )
388
+ self .assertAlmostEqual (y , a0 .Y )
389
+ self .assertAlmostEqual (z + 0.5 , a0 .Z )
390
390
391
391
# Move them back
392
392
self .m .TranslateAtomGroup (self .m .GetAtomList (), 0 , 0 , - 0.5 )
393
- self .assertAlmostEquals (x , a0 .X )
394
- self .assertAlmostEquals (y , a0 .Y )
395
- self .assertAlmostEquals (z , a0 .Z )
393
+ self .assertAlmostEqual (x , a0 .X )
394
+ self .assertAlmostEqual (y , a0 .Y )
395
+ self .assertAlmostEqual (z , a0 .Z )
396
396
397
397
# Rotate the atoms
398
398
@@ -406,9 +406,9 @@ def testManipulation(self):
406
406
for i in range (len (self .m )):
407
407
xyzi = xyz [i ]
408
408
newxyz = numpy .dot (rm , xyzi )
409
- self .assertAlmostEquals (newxyz [0 ], self .m [i ].X , 6 )
410
- self .assertAlmostEquals (newxyz [1 ], self .m [i ].Y , 6 )
411
- self .assertAlmostEquals (newxyz [2 ], self .m [i ].Z , 6 )
409
+ self .assertAlmostEqual (newxyz [0 ], self .m [i ].X , 6 )
410
+ self .assertAlmostEqual (newxyz [1 ], self .m [i ].Y , 6 )
411
+ self .assertAlmostEqual (newxyz [2 ], self .m [i ].Z , 6 )
412
412
413
413
return
414
414
@@ -437,38 +437,38 @@ def testAccessors(self):
437
437
self .assertEqual ("test" , a .GetName ())
438
438
439
439
# Test xyz & occ Get/Set
440
- self .assertAlmostEquals (3.451266498 , a .X , numplaces )
441
- self .assertAlmostEquals (0.685 , a .Y , numplaces )
442
- self .assertAlmostEquals (0 , a .Z , numplaces )
443
- self .assertAlmostEquals (1.0 , a .Occupancy , numplaces )
440
+ self .assertAlmostEqual (3.451266498 , a .X , numplaces )
441
+ self .assertAlmostEqual (0.685 , a .Y , numplaces )
442
+ self .assertAlmostEqual (0 , a .Z , numplaces )
443
+ self .assertAlmostEqual (1.0 , a .Occupancy , numplaces )
444
444
445
445
a .x = 3.40
446
446
a .y = 0.68
447
447
a .z = 0.1
448
448
a .occ = 1.02
449
449
450
- self .assertAlmostEquals (3.40 , a .x , numplaces )
451
- self .assertAlmostEquals (0.68 , a .y , numplaces )
452
- self .assertAlmostEquals (0.1 , a .z , numplaces )
453
- self .assertAlmostEquals (1.02 , a .occ , numplaces )
450
+ self .assertAlmostEqual (3.40 , a .x , numplaces )
451
+ self .assertAlmostEqual (0.68 , a .y , numplaces )
452
+ self .assertAlmostEqual (0.1 , a .z , numplaces )
453
+ self .assertAlmostEqual (1.02 , a .occ , numplaces )
454
454
455
455
# Test GetMolecule. We can't expect the python object to be the same as
456
456
# our molecule above. However, we can verify that it points to the same
457
457
# object.
458
458
m = a .GetMolecule ()
459
- self .assertEquals (m .GetName (), self .m .GetName ())
459
+ self .assertEqual (m .GetName (), self .m .GetName ())
460
460
# Change something with the molecule, and check to see if it appears in
461
461
# self.m
462
462
m .GetAtom ("C1" ).Occupancy = 0.1
463
- self .assertAlmostEquals (0.1 , self .m .GetAtom ("C1" ).Occupancy , numplaces )
463
+ self .assertAlmostEqual (0.1 , self .m .GetAtom ("C1" ).Occupancy , numplaces )
464
464
465
465
# Test IsDummy
466
466
self .assertFalse (a .IsDummy ())
467
467
468
468
# Test GetScatteringPower
469
469
sp = a .GetScatteringPower ()
470
- self .assertEquals ("ScatteringPowerAtom" , sp .GetClassName ())
471
- self .assertEquals ("C" , sp .GetName ())
470
+ self .assertEqual ("ScatteringPowerAtom" , sp .GetClassName ())
471
+ self .assertEqual ("C" , sp .GetName ())
472
472
473
473
# Test Ring Get/Set
474
474
self .assertFalse (a .IsInRing ())
@@ -707,20 +707,20 @@ def testStretchModeBondLength(self):
707
707
sm = StretchModeBondLength (atop , abot , None )
708
708
sm .AddAtom (abot )
709
709
710
- self .assertEquals (sm .mpAtom0 .GetName (), atop .GetName ())
711
- self .assertEquals (sm .mpAtom1 .GetName (), abot .GetName ())
710
+ self .assertEqual (sm .mpAtom0 .GetName (), atop .GetName ())
711
+ self .assertEqual (sm .mpAtom1 .GetName (), abot .GetName ())
712
712
713
713
# Stretch the bond by 5%
714
714
delta = 0.05 * d0
715
715
sm .Stretch (delta )
716
716
717
717
# Make sure this does what we expected
718
718
d1 = GetBondLength (atop , abot )
719
- self .assertAlmostEquals (d0 + delta , d1 , 6 )
719
+ self .assertAlmostEqual (d0 + delta , d1 , 6 )
720
720
721
721
# Note that only the second atom has moved
722
722
dc1 = GetBondLength (ac , atop )
723
- self .assertAlmostEquals (dc0 , dc1 )
723
+ self .assertAlmostEqual (dc0 , dc1 )
724
724
725
725
return
726
726
@@ -752,9 +752,9 @@ def testStretchModeBondAngle(self):
752
752
sm = StretchModeBondAngle (a1 , ac , a2 , None )
753
753
sm .AddAtom (a2 )
754
754
755
- self .assertEquals (sm .mpAtom0 .GetName (), a1 .GetName ())
756
- self .assertEquals (sm .mpAtom1 .GetName (), ac .GetName ())
757
- self .assertEquals (sm .mpAtom2 .GetName (), a2 .GetName ())
755
+ self .assertEqual (sm .mpAtom0 .GetName (), a1 .GetName ())
756
+ self .assertEqual (sm .mpAtom1 .GetName (), ac .GetName ())
757
+ self .assertEqual (sm .mpAtom2 .GetName (), a2 .GetName ())
758
758
759
759
760
760
# Stretch the angle by 5%
@@ -763,7 +763,7 @@ def testStretchModeBondAngle(self):
763
763
764
764
# Make sure this does what we expected
765
765
angle1 = GetBondAngle (a1 , ac , a2 )
766
- self .assertAlmostEquals (angle0 + delta , angle1 , 6 )
766
+ self .assertAlmostEqual (angle0 + delta , angle1 , 6 )
767
767
768
768
return
769
769
@@ -797,16 +797,16 @@ def testStretchModeTorsion(self):
797
797
# Add the last atom so it can rotate
798
798
sm .AddAtom (a2 )
799
799
800
- self .assertEquals (sm .mpAtom1 .GetName (), ac0 .GetName ())
801
- self .assertEquals (sm .mpAtom2 .GetName (), ac1 .GetName ())
800
+ self .assertEqual (sm .mpAtom1 .GetName (), ac0 .GetName ())
801
+ self .assertEqual (sm .mpAtom2 .GetName (), ac1 .GetName ())
802
802
803
803
# Stretch the angle by 5%
804
804
delta = 0.25 * angle0
805
805
sm .Stretch (delta )
806
806
807
807
# Make sure this does what we expected
808
808
angle1 = GetDihedralAngle (a1 , ac0 , ac1 , a2 )
809
- self .assertAlmostEquals (angle0 + delta , angle1 , 6 )
809
+ self .assertAlmostEqual (angle0 + delta , angle1 , 6 )
810
810
811
811
return
812
812
0 commit comments