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')
