def test_if_has_valid_data_config(self):
     experimenter = Experimenter(data_config=['a', 'b'])
     self.assertFalse(experimenter.has_valid_data_config())
     experimenter.data_config = {'A': 'a', 'B': 'b'}
     self.assertFalse(experimenter.has_valid_data_config())
     experimenter.data_config = {'Train': 'a', 'B': 'b'}
     self.assertFalse(experimenter.has_valid_data_config())
     experimenter.data_config = {'Train': 'a', 'Test': 'b'}
     self.assertFalse(experimenter.has_valid_data_config())
     experimenter.data_config = {'train': {'a': 1}, 'test': {'b': 2}}
     self.assertFalse(experimenter.has_valid_data_config())
     experimenter.data_config = {'train': {'a': 1}, 'test': {'b': 2}, 'select_sql': """SELECT hello FROM crap"""}
     self.assertTrue(experimenter.has_valid_data_config())
     experimenter.data_config = {'train': {'tag1': '2344'}, 'test': {'tag1': '2344'},
                                 'select_sql': 'SELECT * FROM {tag2}'}
     self.assertFalse(experimenter.has_valid_data_config())
     experimenter.data_config = {'train': {'tag1': '2344'}, 'test': {'tag1': '2344'},
                                 'select_sql': 'SELECT * FROM {tag1}'}
     self.assertTrue(experimenter.has_valid_data_config())
Ejemplo n.º 2
0
 def test_if_has_valid_data_config(self):
     experimenter = Experimenter(data_config=['a', 'b'])
     self.assertFalse(experimenter.has_valid_data_config())
     experimenter.data_config = {'A': 'a', 'B': 'b'}
     self.assertFalse(experimenter.has_valid_data_config())
     experimenter.data_config = {'Train': 'a', 'B': 'b'}
     self.assertFalse(experimenter.has_valid_data_config())
     experimenter.data_config = {'Train': 'a', 'Test': 'b'}
     self.assertFalse(experimenter.has_valid_data_config())
     experimenter.data_config = {'train': {'a': 1}, 'test': {'b': 2}}
     self.assertFalse(experimenter.has_valid_data_config())
     experimenter.data_config = {
         'train': {
             'a': 1
         },
         'test': {
             'b': 2
         },
         'select_sql': """SELECT hello FROM crap"""
     }
     self.assertTrue(experimenter.has_valid_data_config())
     experimenter.data_config = {
         'train': {
             'tag1': '2344'
         },
         'test': {
             'tag1': '2344'
         },
         'select_sql': 'SELECT * FROM {tag2}'
     }
     self.assertFalse(experimenter.has_valid_data_config())
     experimenter.data_config = {
         'train': {
             'tag1': '2344'
         },
         'test': {
             'tag1': '2344'
         },
         'select_sql': 'SELECT * FROM {tag1}'
     }
     self.assertTrue(experimenter.has_valid_data_config())