Ejemplo n.º 1
0
 def test_build_data_config(self):
     experimenter = Experimenter(
         data_config={
             'train': {
                 'table': 'train_data'
             },
             'test': {
                 'table': 'test_data'
             },
             'select_sql': 'SELECT * FROM {table}'
         })
     experimenter.build_data_config()
     self.assertEqual(experimenter.data_config['train'],
                      'SELECT * FROM train_data')
     self.assertEqual(experimenter.data_config['test'],
                      'SELECT * FROM test_data')
 def test_build_data_config(self):
     experimenter = Experimenter(data_config={'train': {'table': 'train_data'}, 'test': {'table': 'test_data'},
                                              'select_sql': 'SELECT * FROM {table}'})
     experimenter.build_data_config()
     self.assertEqual(experimenter.data_config['train'], 'SELECT * FROM train_data')
     self.assertEqual(experimenter.data_config['test'], 'SELECT * FROM test_data')