fixed benchmark for fragment timing

(useful for MDAnalysis#2376)
SanjanaSogimatt · Nov 14, 2019 · 4b82988 · 4b82988
1 parent e110108
commit 4b82988
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/benchmarks/benchmarks/ag_methods.py b/benchmarks/benchmarks/ag_methods.py
@@ -4,8 +4,8 @@
 import numpy as np
 
 try:
-    from MDAnalysisTests.datafiles import (GRO, TPR, XTC, 
-                                           PSF, DCD, 
+    from MDAnalysisTests.datafiles import (GRO, TPR, XTC,
+                                           PSF, DCD,
                                            TRZ_psf, TRZ)
     from MDAnalysis.exceptions import NoDataError
 except:
@@ -190,7 +190,7 @@ def time_translate(self, num_atoms):
         coordinates.
         """
         self.ag.translate([0,0.5,1])
-        
+
     def time_union(self, num_atoms):
         """Benchmark union operation
         on atomgroups.
@@ -202,7 +202,7 @@ def time_wrap(self, num_atoms):
         atomgroup with default params.
         """
         self.ag.wrap()
-        
+
 
 
 class AtomGroupAttrsBench(object):
@@ -232,7 +232,7 @@ def time_atoms(self, num_atoms):
 
     def time_dihedral(self, num_atoms):
         """Benchmark Dihedral object
-        creation time. Requires ag of 
+        creation time. Requires ag of
         size 4.
         """
         self.ag[:4].dihedral
@@ -336,5 +336,5 @@ class FragmentFinding(object):
     def setup(self, universe):
         self.u = MDAnalysis.Universe(*universe)
 
-    def test_find_fragments(self, universe):
+    def time_find_fragments(self, universe):
         frags = self.u.atoms.fragments