def test_cast_array(self):
"""Make sure array are cast to int and returned as array of values"""
dim = Integer('yolo', 'uniform', -3, 4)
assert np.all(dim.cast(np.array(['1', '2'])) == np.array([1, 2]))
def test_cast_list(self):
"""Make sure list are cast to int and returned as list of values"""
dim = Integer('yolo', 'uniform', -3, 4)
assert dim.cast(['1', '2']) == [1, 2]
def test_cast_list(self):
"""Make sure list are cast to int and returned as list of values"""
dim = Integer("yolo", "uniform", -3, 5)
assert dim.cast(["1", "2"]) == [1, 2]
def test_cast_array(self):
"""Make sure array are cast to int and returned as array of values"""
dim = Integer("yolo", "uniform", -3, 5)
assert np.all(dim.cast(np.array(["1", "2"])) == np.array([1, 2]))
def test_cast_borders(self):
"""Make sure cast to int returns correct borders"""
dim = Integer("yolo", "uniform", -3, 5)
assert dim.cast(-3.0) == -3
assert dim.cast(2.0) == 2
