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