updated tests

MDAnalysis · Jul 6, 2018 · 5ed599f · 5ed599f
1 parent f3e74c9
commit 5ed599f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/testsuite/MDAnalysisTests/transformations/test_rotate.py b/testsuite/MDAnalysisTests/transformations/test_rotate.py
@@ -120,7 +120,7 @@ def test_rotateby_atomgroup_cog_pbc(rotate_universes):
     center_pos = selection.center_of_geometry(pbc=True)
     matrix = rotation_matrix(np.deg2rad(angle), vector, center_pos)[:3, :3]
     ref.positions = np.dot(ref.positions, matrix)
-    transformed = rotateby(angle, vector, ag=selection, center='geometry', pbc=True)(trans) 
+    transformed = rotateby(angle, vector, ag=selection, center='geometry', wrap=True)(trans) 
     assert_array_almost_equal(transformed.positions, ref.positions, decimal=6)
 
 def test_rotateby_atomgroup_com_pbc(rotate_universes):
@@ -136,7 +136,7 @@ def test_rotateby_atomgroup_com_pbc(rotate_universes):
     center_pos = selection.center_of_mass(pbc=True)
     matrix = rotation_matrix(np.deg2rad(angle), vector, center_pos)[:3, :3]
     ref.positions = np.dot(ref.positions, matrix)
-    transformed = rotateby(angle, vector, ag=selection, center='mass', pbc=True)(trans) 
+    transformed = rotateby(angle, vector, ag=selection, center='mass', wrap=True)(trans) 
     assert_array_almost_equal(transformed.positions, ref.positions, decimal=6)
 
 def test_rotateby_bad_ag(rotate_universes):
@@ -169,7 +169,7 @@ def test_rotateby_bad_pbc(rotate_universes):
     angle = 90
     vector = [0, 0, 1]
     with pytest.raises(ValueError): 
-        rotateby(angle, vector, ag = ag, pbc=True)(ts)
+        rotateby(angle, vector, ag = ag, wrap=True)(ts)
 
 def test_rotateby_bad_center(rotate_universes):
     # this universe as a box size zero