removed test of group-wise calc and replaced it with option 2 test

halotools/utils/tests/test_rotate_vector_collection.py

@@ -13,7 +13,7 @@
 
 def test_rotation_1():
     """
-    single rotation matrix + single set of points
+    test option 1: single rotation matrix + set of points
     """
 
     # create a single rotation matrix
@@ -24,7 +24,7 @@ def test_rotation_1():
     rot_m = rotation_matrices_from_vectors(v1,v2)
     rot = rot_m[0]
 
-    # create a single set of vectors
+    # create a set of vectors
     npts = 1000
     ndim = 3
     v3 = np.random.random((npts,ndim))
@@ -34,22 +34,20 @@ def test_rotation_1():
     assert np.shape(v4)==(npts, ndim)
 
 
-def test_rotation_3():
+def test_rotation_2():
     """
-    nset of rotation matrices + nset of npts of points
+    test option 2: n rotation matrices + n points
     """
 
-    nsets = 2
+    npts = 1000
     ndim = 3
-    v1 = np.random.random((nsets,ndim))
-    v2 = np.random.random((nsets,ndim))
+    v1 = np.random.random((npts,ndim))
+    v2 = np.random.random((npts,ndim))
 
     rot = rotation_matrices_from_vectors(v1,v2)
 
-    npts = 1000
-    ndim = 3
-    v3 = np.random.random((nsets, npts, ndim))
+    v3 = np.random.random((npts, ndim))
 
     v4 = rotate_vector_collection(rot, v3)
 
-    assert np.shape(v4)==(nsets, npts, ndim)
+    assert np.shape(v4)==(npts, ndim)