def test_check_value_wrong_type():
with pytest.raises(ValueError) as e:
check_value({}, (3, ))
assert "Expected an array with shape (3,); got dict" in str(e.value)
class Value:
def __init__(self):
self.shape = None
with pytest.raises(ValueError) as e:
check_value(Value(), (3, ))
assert "Expected an array with shape (3,); got Value" in str(e.value)
def test_check_value_valid_scalar():
check_value(np.int64(3), ())
def test_check_value_with_invalid_shape_raises_expected_error():
with pytest.raises(ValueError) as e:
check_value(np.zeros(3), (3.0, ))
assert "Expected shape dimensions to be int" in str(e.value)
def test_check_value_none_named():
with pytest.raises(ValueError) as e:
check_value(None, (3, ), name="input_value")
assert "input_value must be an array with shape (3,); got None" in str(
e.value)
def test_check_value_wrong_shape_wildcard_named():
with pytest.raises(ValueError) as e:
check_value(np.zeros((5, 4)), (-1, 3), name="input_value")
assert "input_value must be an array with shape (-1, 3); got (5, 4)" in str(
e.value)
def test_check_value_wrong_shape_named():
with pytest.raises(ValueError) as e:
check_value(np.zeros(4), (3, ), name="input_value")
assert "input_value must be an array with shape (3,); got (4,)" in str(
e.value)
def test_check_value_valid_wildcard_named():
assert check_value(np.zeros((5, 3)), (-1, 3), name="input_value") == 5
assert check_value(np.zeros((5, 3)), (5, -1), name="input_value") == 3
def test_check_value_valid_named():
check_value(np.zeros(3), (3, ), name="input_value")
def test_check_value_valid():
check_value(np.zeros(3), (3, ))
def test_check_value_none():
with pytest.raises(ValueError) as e:
check_value(None, (3, ))
assert "Expected an array with shape (3,); got None" in str(e.value)
def test_check_value_wrong_shape_wildcard():
with pytest.raises(ValueError) as e:
check_value(np.zeros((5, 4)), (-1, 3))
assert "Expected an array with shape (-1, 3); got (5, 4)" in str(e.value)
def test_check_value_wrong_shape():
with pytest.raises(ValueError) as e:
check_value(np.zeros(4), (3, ))
assert "Expected an array with shape (3,); got (4,)" in str(e.value)
def test_check_value_valid_wildcard():
assert check_value(np.zeros((5, 3)), (-1, 3)) == 5
assert check_value(np.zeros((5, 3)), (5, -1)) == 3
assert check_value(np.zeros((5, 3, 2)), (-1, 3, -1)) == (5, 2)
