t/moe/tasktypes/dummy.py

   1 """
   2
   3 Dummy task type.
   4
   5 Adds several info-printing hooks to both pipelines.
   6 Runs the test pipeline for each TEST
   7 """
   8
   9 import moe.config
  10
  11 def init(e):
  12
  13     def hook_m_50(e):
  14         e.log.info('Here should be compiling')
  15
  16     e.main_pipe.insert(50, 'dummy tasktype hook_m_50', hook_m_50)
  17
  18     def hook_m_60(e):
  19         e.log.info('Here we run test pipeline')
  20         tests = e['TESTS'].split()
  21         for t in tests:
  22              e.log.info('Running test %s' % t)
  23              e.log.user.info('TEST %s ...' % t)
  24
  25              op = moe.config.Operation('SET', None, moe.config.ConfigExpression([t]), level=70, source='<dummy tasktype>')
  26              testvar = e.config.lookup('TEST')
  27              # Add and remove operation setting 'TEST="..."'
  28              testvar.add_operation(op)
  29              e.test_pipe.run(e=e)
  30              testvar.remove_operation(op)
  31
  32     e.main_pipe.insert(60, 'dummy tasktype hook_m_60', hook_m_60)
  33
  34     def hook_t_30(e):
  35         e.log.info("Maybe we should do something? Nah...")
  36
  37     e.test_pipe.insert(30, 'dummy tasktype hook_t_30', hook_t_30)