def test_load_orion_name(self): orion = functional._load_orion('dummy') assert isinstance(orion, Orion) assert orion._pipeline == 'dummy' assert not orion._fitted assert orion._hyperparameters is None
def test_load_orion_pickle(self, tmpdir): path = os.path.join(tmpdir, 'orion.pkl') self.orion.save(path) orion = functional._load_orion(path) assert orion is not self.orion assert orion == self.orion
def test_load_orion_dict(self): pipeline = load_pipeline('dummy') orion = functional._load_orion(pipeline) assert isinstance(orion, Orion) assert orion._pipeline == pipeline assert not orion._fitted assert orion._hyperparameters is None
def test_load_orion_json_path(self, tmpdir): pipeline = load_pipeline('dummy') path = os.path.join(tmpdir, 'pipeline.json') with open(path, 'w') as json_file: json.dump(pipeline, json_file) orion = functional._load_orion(path) assert isinstance(orion, Orion) assert orion._pipeline == path assert not orion._fitted assert orion._hyperparameters is None
def test_load_orion_hyperparams(self): hyperparams = { "orion.primitives.detectors.ThresholdDetector#1": { "ratio": 0.9 } } orion = functional._load_orion('dummy', hyperparams) assert isinstance(orion, Orion) assert orion._pipeline == 'dummy' assert not orion._fitted assert orion._hyperparameters == hyperparams
def test_load_orion_orion(self): orion = functional._load_orion(self.orion) assert orion is self.orion
def test_load_orion_invalid(self): with pytest.raises(ValueError): functional._load_orion('invalid')