def test_cast_list(self):
"""Make sure list are cast to categories and returned as list"""
categories = {'asdfa': 0.1, 2: 0.2, 3.0: 0.3, 'lalala': 0.4}
dim = Categorical('yolo', categories)
assert dim.cast(['asdfa']) == ['asdfa']
assert dim.cast(['2']) == [2]
assert dim.cast(['3.0']) == [3.0]
def test_cast_list(self):
"""Make sure list are cast to categories and returned as list"""
categories = {"asdfa": 0.1, 2: 0.2, 3.0: 0.3, "lalala": 0.4}
dim = Categorical("yolo", categories)
assert dim.cast(["asdfa"]) == ["asdfa"]
assert dim.cast(["2"]) == [2]
assert dim.cast(["3.0"]) == [3.0]
def test_cast_bad_category(self):
"""Make sure array are cast to int and returned as array of values"""
categories = list(range(10))
dim = Categorical('yolo', categories, shape=2)
sample = np.array(['asdfa', '1'], dtype=np.object)
with pytest.raises(ValueError) as exc:
dim.cast(sample)
assert "Invalid category: asdfa" in str(exc.value)
def test_cast_list_multidim(self):
"""Make sure array are cast to int and returned as array of values"""
categories = list(range(10))
categories[0] = 'asdfa'
categories[2] = 'lalala'
dim = Categorical('yolo', categories, shape=2)
sample = ['asdfa', '1'] # np.array(['asdfa', '1'], dtype=np.object)
assert dim.cast(sample) == ['asdfa', 1]
def test_cast_array_multidim(self):
"""Make sure array are cast to int and returned as array of values"""
categories = list(range(10))
categories[0] = "asdfa"
categories[2] = "lalala"
dim = Categorical("yolo", categories, shape=2)
sample = np.array(["asdfa", "1"], dtype=np.object)
assert np.all(dim.cast(sample) == np.array(["asdfa", 1], dtype=np.object))
