提交 383600bc authored 作者: Ricardo's avatar Ricardo 提交者: Brandon T. Willard

Add get_scalar_constant_value test for None and NoneConst

上级 9768905a
......@@ -22,6 +22,7 @@ from aesara.graph.op import Op
from aesara.misc.safe_asarray import _asarray
from aesara.raise_op import Assert
from aesara.scalar import autocast_float, autocast_float_as
from aesara.tensor import NoneConst
from aesara.tensor.basic import (
Alloc,
AllocDiag,
......@@ -3524,6 +3525,19 @@ class TestGetScalarConstantValue:
e = extract_constant(c)
assert e == 3, (c, d, e)
@pytest.mark.parametrize("only_process_constants", (True, False))
def test_None_and_NoneConst(self, only_process_constants):
with pytest.raises(NotScalarConstantError):
get_scalar_constant_value(
None, only_process_constants=only_process_constants
)
assert (
get_scalar_constant_value(
NoneConst, only_process_constants=only_process_constants
)
is None
)
def test_complex_mod_failure():
# Make sure % fails on complex numbers.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论