tweaking tests

Author: realead

asv_bench/benchmarks/series_methods.py

@@ -27,24 +27,34 @@ def time_constructor(self, data):
 class IsIn(object):
 
     goal_time = 0.2
-    params = ['int64', 'object', 'float64']
+    params = ['int64', 'object']
     param_names = ['dtype']
 
     def setup(self, dtype):
         self.s = Series(np.random.randint(1, 10, 100000)).astype(dtype)
         self.values = [1, 2]
-        self.small = Series(np.random.randint(1, 10, 10)).astype(dtype)
-        self.many_different_values = np.arange(10**6).astype(dtype)
-        self.few_different_values = np.zeros(10**6).astype(dtype)
 
     def time_isin(self, dtypes):
         self.s.isin(self.values)
 
-    def time_isin_many_different(self, dtypes):
+
+class IsInFloat64(object):
+
+    def setup(self):
+        self.small = Series(np.random.randint(1, 10, 10)).astype(dtype=np.float64)
+        self.many_different_values = np.arange(10**6, dtype=np.float64)
+        self.few_different_values = np.zeros(10**7, dtype=np.float64)      
+        self.only_nans_values = np.full(10**7, np.nan, dtype=np.float64)
+
+    def time_isin_many_different(self):
         # runtime is dominated by creation of the lookup-table
         self.small.isin(self.many_different_values)
 
-    def time_isin_few_different(self, dtypes):
+    def time_isin_few_different(self):
+        # runtime is dominated by creation of the lookup-table
+        self.small.isin(self.few_different_values)
+
+    def time_isin_nan_values(self):
         # runtime is dominated by creation of the lookup-table
         self.small.isin(self.few_different_values)