def test_interval_is_banned(self):
"""Check that calling `Categorical.interval` raises `RuntimeError`."""
categories = {'asdfa': 0.1, 2: 0.2, 3: 0.3, 4: 0.4}
dim = Categorical('yolo', categories, shape=2)
with pytest.raises(RuntimeError) as exc:
dim.interval()
assert 'not ordered' in str(exc.value)
def choices(self, *args, **kwargs):
"""Create a `Categorical` dimension."""
name = self.name
try:
if isinstance(args[0], (dict, list)):
return Categorical(name, *args, **kwargs)
except IndexError as exc:
raise TypeError(
"Parameter '{}': "
"Expected argument with categories.".format(name)) from exc
return Categorical(name, args, **kwargs)
def test_probabilities_are_ok(self, seed):
"""Test that the probabilities given are legit using law of big numbers."""
bins = defaultdict(int)
probs = (0.1, 0.2, 0.3, 0.4)
categories = ('asdfa', '2', '3', '4')
categories = OrderedDict(zip(categories, probs))
dim = Categorical('yolo', categories)
for _ in range(500):
sample = dim.sample(seed=seed)[0]
bins[sample] += 1
for keys in bins.keys():
bins[keys] /= float(500)
for key, value in categories.items():
assert abs(bins[key] - value) < 0.01
def test_that_objects_types_are_ok(self):
"""Check that output samples are of the correct type.
Don't let numpy mess with their automatic type inference.
"""
categories = {'asdfa': 0.1, 2: 0.2, 3: 0.3, 'lalala': 0.4}
dim = Categorical('yolo', categories)
assert '2' not in dim
assert 2 in dim
assert 'asdfa' in dim
dim = Categorical('yolo', categories, shape=(2, ))
assert ['2', 'asdfa'] not in dim
assert [2, 'asdfa'] in dim
def test_contains_wrong_shape(self):
"""Check correct category but wrongly shaped array."""
categories = {'asdfa': 0.1, 2: 0.2, 3: 0.3, 4: 0.4}
dim = Categorical('yolo', categories, shape=2)
assert 3 not in dim
assert ('asdfa', 2) in dim
def test_repr_too_many_cats(self):
"""Check ellipsis on str/repr of too many categories."""
categories = tuple(range(10))
dim = Categorical('yolo', categories, shape=2)
assert str(dim) == "Categorical(name=yolo, " \
"prior={0: 0.10, 1: 0.10, ..., 8: 0.10, 9: 0.10}, " \
"shape=(2,))"
def test_with_dict(self, seed):
"""Test Categorical.__init__ with a dictionary."""
probs = (0.1, 0.2, 0.3, 0.4)
categories = ('asdfa', 2, 3, 4)
dim = Categorical('yolo', OrderedDict(zip(categories, probs)))
samples = dim.sample(seed=seed)
assert len(samples) == 1
assert samples[0] == 2
assert dim._probs == probs
assert categories == dim.categories
assert 2 in dim
assert 0 not in dim
assert dim.name == 'yolo'
assert dim.type == 'categorical'
assert dim.shape == ()
def space():
"""Construct a simple space with every possible kind of Dimension."""
space = Space()
categories = {'asdfa': 0.1, 2: 0.2, 3: 0.3, 4: 0.4}
dim = Categorical('yolo', categories, shape=2)
space.register(dim)
dim = Integer('yolo2', 'uniform', -3, 6)
space.register(dim)
dim = Real('yolo3', 'alpha', 0.9)
space.register(dim)
return space
def test_with_tuple(self, seed):
"""Test Categorical.__init__ with a tuple."""
categories = ('asdfa', 2)
dim = Categorical('yolo', categories)
samples = dim.sample(seed=seed)
assert len(samples) == 1
assert samples[0] == 'asdfa'
assert dim._probs == (0.5, 0.5)
assert categories == dim.categories
assert 2 in dim
assert 3 not in dim
assert str(
dim
) == "Categorical(name=yolo, prior={asdfa: 0.50, 2: 0.50}, shape=())"
assert dim.name == 'yolo'
assert dim.type == 'categorical'
assert dim.shape == ()
def test_interval(self):
"""Check whether interval is cool."""
space = Space()
probs = (0.1, 0.2, 0.3, 0.4)
categories = ('asdfa', 2, 3, 4)
dim = Categorical('yolo', OrderedDict(zip(categories, probs)), shape=2)
space.register(dim)
dim = Integer('yolo2', 'uniform', -3, 6)
space.register(dim)
dim = Real('yolo3', 'norm', 0.9)
space.register(dim)
assert space.interval() == [categories, (-3, 3), (-np.inf, np.inf)]
def test_sample(self, seed):
"""Check whether sampling works correctly."""
space = Space()
probs = (0.1, 0.2, 0.3, 0.4)
categories = ('asdfa', 2, 3, 4)
dim1 = Categorical('yolo',
OrderedDict(zip(categories, probs)),
shape=(2, 2))
space.register(dim1)
dim2 = Integer('yolo2', 'uniform', -3, 6)
space.register(dim2)
dim3 = Real('yolo3', 'norm', 0.9)
space.register(dim3)
point = space.sample(seed=seed)
test_point = [
(dim1.sample()[0], dim2.sample()[0], dim3.sample()[0]),
]
assert len(point) == len(test_point) == 1
assert len(point[0]) == len(test_point[0]) == 3
assert np.all(point[0][0] == test_point[0][0])
assert point[0][1] == test_point[0][1]
assert point[0][2] == test_point[0][2]
points = space.sample(2, seed=seed)
points1 = dim1.sample(2)
points2 = dim2.sample(2)
points3 = dim3.sample(2)
test_points = [(points1[0], points2[0], points3[0]),
(points1[1], points2[1], points3[1])]
assert len(points) == len(test_points) == 2
for i in range(2):
assert len(points[i]) == len(test_points[i]) == 3
assert np.all(points[i][0] == test_points[i][0])
assert points[i][1] == test_points[i][1]
assert points[i][2] == test_points[i][2]
def test_getitem(self):
"""Test getting dimensions from space."""
space = Space()
probs = (0.1, 0.2, 0.3, 0.4)
categories = ('asdfa', 2, 3, 4)
dim = Categorical('yolo', OrderedDict(zip(categories, probs)), shape=2)
space.register(dim)
dim = Integer('yolo2', 'uniform', -3, 6)
space.register(dim)
dim = Real('yolo3', 'norm', 0.9)
space.register(dim)
assert space['yolo'].type == 'categorical'
assert space[0].type == 'categorical'
with pytest.raises(KeyError):
space['asdf']
with pytest.raises(IndexError):
space[3]
def test_register_and_contain(self):
"""Register bunch of dimensions, check if points/name are in space."""
space = Space()
assert 'yolo' not in space
assert (('asdfa', 2), 0, 3.5) not in space
categories = {'asdfa': 0.1, 2: 0.2, 3: 0.3, 4: 0.4}
dim = Categorical('yolo', categories, shape=2)
space.register(dim)
dim = Integer('yolo2', 'uniform', -3, 6)
space.register(dim)
dim = Real('yolo3', 'norm', 0.9)
space.register(dim)
assert 'yolo' in space
assert 'yolo2' in space
assert 'yolo3' in space
assert (('asdfa', 2), 0, 3.5) in space
assert (('asdfa', 2), 7, 3.5) not in space
def test_bad_probabilities(self):
"""User provided bad probabilities."""
categories = {'asdfa': 0.05, 2: 0.2, 3: 0.3, 4: 0.4}
with pytest.raises(ValueError):
Categorical('yolo', categories, shape=2)