Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_family_class():
prior = Prior('CheeseWhiz', holes=0, taste=-10)
family = Family('cheese', prior, link='ferment', parent='holes')
for name in ['name', 'prior', 'link', 'parent']:
assert hasattr(family, name)
def test_prior_retrieval():
config_file = join(dirname(__file__), 'data', 'sample_priors.json')
pf = PriorFactory(config_file)
prior = pf.get(dist='asiago')
assert prior.name == 'Asiago'
assert isinstance(prior, Prior)
assert prior.args['hardness'] == 10
with pytest.raises(KeyError):
assert prior.args['holes'] == 4
family = pf.get(family='hard')
assert isinstance(family, Family)
assert family.link == 'grate'
backup = family.prior.args['backup']
assert isinstance(backup, Prior)
assert backup.args['flavor'] == 10000
prior = pf.get(term='yellow')
assert prior.name == 'Swiss'
"""
if dist is not None:
if dist not in self.dists:
raise ValueError("'%s' is not a valid distribution name." % dist)
return self._get_prior(self.dists[dist])
elif term is not None:
if term not in self.terms:
raise ValueError("'%s' is not a valid term type." % term)
return self._get_prior(self.terms[term])
elif family is not None:
if family not in self.families:
raise ValueError("'%s' is not a valid family name." % family)
_f = self.families[family]
prior = self._get_prior(_f["dist"])
return Family(family, prior, _f["link"], _f["parent"])