patch 4.0

unittest/sql/optimizer/cost_model_utils/op_generator.py

@@ -0,0 +1,59 @@
+from cost_test_conf import Config
+from mylog.mylog import MyLogger
+import subprocess as sp
+
+
+def init_func(self, conf):
+    self.conf = conf
+
+
+def get_bench_cmd(self):
+    cmd = './cost_model_util ' + self.conf.gen_params()
+    return cmd
+
+def get_data_preprocess_cmd(self):
+    cmd = 'python preprocess.py -i {0} -o {1} -d'.format(
+        self.__class__.__name__ + '_result',
+        self.__class__.__name__ + '_result_final'
+    )
+    return cmd
+
+def do_bench(self):
+    MyLogger.info(self.conf)
+    cmd = self.get_bench_cmd()
+    MyLogger.info(cmd)
+    sp.check_call(cmd, shell=True)
+    data_cmd = self.get_data_preprocess_cmd()
+    sp.check_call(data_cmd, shell=True)
+
+
+class op_generator(object):
+    op_dict = {}
+    '''
+    name if type is not None name = operatorname + test_type_name
+    '''
+
+    @staticmethod
+    def gen_operator(name):
+        if op_generator.op_dict.has_key(name):
+            return op_generator.op_dict[name]
+        else:
+            cls = type(name, (object,), {'__init__': init_func, 'do_bench': do_bench,
+                                         'get_bench_cmd': get_bench_cmd,
+                                         'get_data_preprocess_cmd': get_data_preprocess_cmd})
+            op_generator.op_dict[name] = cls
+            return cls
+
+
+if __name__ == '__main__':
+    ##mat related conf
+    material_cls = op_generator.gen_operator('material')
+    conf = Config()
+    conf.u_to_test_op_c = 'material'
+    conf.is_not_running_as_unittest_c = True
+    conf.schema_file_c = 'c10k1.schema'
+    conf.row_count_c = 1000
+    conf.input_projector_count_c = 1
+
+    material_op = material_cls(conf)
+    material_op.do_bench()