def test_has_valid_model_config(self):
experimenter = Experimenter()
experimenter.model_config = [{'ignore': [], 'features': {'feat1': lambda x: x}, 'inherit': []}]
self.assertFalse(experimenter.has_valid_model_config())
experimenter.model_config = {'base_model': {'ignore': [], 'features': {'feat1': lambda x: x}, 'inherit': []}}
self.assertTrue(experimenter.has_valid_model_config())
experimenter.model_config = {
'derived_model': {'ignore': [], 'features': {'feat1': lambda x: x}, 'inherit': ['base_model']}}
self.assertFalse(experimenter.has_valid_model_config())
experimenter.model_config = {'base_model': {'ignore': [], 'features': {'feat1': lambda x: x}, 'inherit': []},
'derived_model': {'ignore': [], 'features': {'feat1': lambda x: x},
'inherit': ['base_model']}}
self.assertTrue(experimenter.has_valid_model_config())
def test_has_valid_model_config(self):
experimenter = Experimenter()
experimenter.model_config = [{
'ignore': [],
'features': {
'feat1': lambda x: x
},
'inherit': []
}]
self.assertFalse(experimenter.has_valid_model_config())
experimenter.model_config = {
'base_model': {
'ignore': [],
'features': {
'feat1': lambda x: x
},
'inherit': []
}
}
self.assertTrue(experimenter.has_valid_model_config())
experimenter.model_config = {
'derived_model': {
'ignore': [],
'features': {
'feat1': lambda x: x
},
'inherit': ['base_model']
}
}
self.assertFalse(experimenter.has_valid_model_config())
experimenter.model_config = {
'base_model': {
'ignore': [],
'features': {
'feat1': lambda x: x
},
'inherit': []
},
'derived_model': {
'ignore': [],
'features': {
'feat1': lambda x: x
},
'inherit': ['base_model']
}
}
self.assertTrue(experimenter.has_valid_model_config())
def test_build_model_config(self):
experimenter = Experimenter()
func = lambda x: x
experimenter.model_config = {'base_model': {'ignore': [], 'features': {'feat1': func}, 'inherit': []}}
experimenter.build_model_config()
self.assertEquals(experimenter.model_config, {'base_model': {'feat1': func}})
experimenter.model_config = {'base_model': {'ignore': [], 'features': {'feat1': func}, 'inherit': []},
'derived_model': {'ignore': [], 'features': {}, 'inherit': ['base_model']}}
experimenter.build_model_config()
self.assertEquals(experimenter.model_config, {'base_model': {'feat1': func}, 'derived_model': {'feat1': func}})
experimenter.model_config = {'base_model': {'ignore': [], 'features': {'feat1': func}, 'inherit': []},
'derived_model': {'ignore': ['feat1'], 'features': {'feat2': None},
'inherit': ['base_model']}}
experimenter.build_model_config()
self.assertEquals(experimenter.model_config, {'base_model': {'feat1': func}, 'derived_model': {'feat2': None}})
experimenter.model_config = {
'base_model': {'ignore': [], 'features': {'feat1': func}, 'inherit': ['derived_model']},
'derived_model': {'ignore': ['feat1'], 'features': {'feat2': None}, 'inherit': ['base_model']}}
experimenter.build_model_config()
self.assertEquals(experimenter.model_config, {'base_model': {'feat2': None}, 'derived_model': {'feat2': None}})
def test_build_model_config(self):
experimenter = Experimenter()
func = lambda x: x
experimenter.model_config = {
'base_model': {
'ignore': [],
'features': {
'feat1': func
},
'inherit': []
}
}
experimenter.build_model_config()
self.assertEquals(experimenter.model_config,
{'base_model': {
'feat1': func
}})
experimenter.model_config = {
'base_model': {
'ignore': [],
'features': {
'feat1': func
},
'inherit': []
},
'derived_model': {
'ignore': [],
'features': {},
'inherit': ['base_model']
}
}
experimenter.build_model_config()
self.assertEquals(experimenter.model_config, {
'base_model': {
'feat1': func
},
'derived_model': {
'feat1': func
}
})
experimenter.model_config = {
'base_model': {
'ignore': [],
'features': {
'feat1': func
},
'inherit': []
},
'derived_model': {
'ignore': ['feat1'],
'features': {
'feat2': None
},
'inherit': ['base_model']
}
}
experimenter.build_model_config()
self.assertEquals(experimenter.model_config, {
'base_model': {
'feat1': func
},
'derived_model': {
'feat2': None
}
})
experimenter.model_config = {
'base_model': {
'ignore': [],
'features': {
'feat1': func
},
'inherit': ['derived_model']
},
'derived_model': {
'ignore': ['feat1'],
'features': {
'feat2': None
},
'inherit': ['base_model']
}
}
experimenter.build_model_config()
self.assertEquals(experimenter.model_config, {
'base_model': {
'feat2': None
},
'derived_model': {
'feat2': None
}
})
