提交 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 ...@@ -22,6 +22,7 @@ from aesara.graph.op import Op
from aesara.misc.safe_asarray import _asarray from aesara.misc.safe_asarray import _asarray
from aesara.raise_op import Assert from aesara.raise_op import Assert
from aesara.scalar import autocast_float, autocast_float_as from aesara.scalar import autocast_float, autocast_float_as
from aesara.tensor import NoneConst
from aesara.tensor.basic import ( from aesara.tensor.basic import (
Alloc, Alloc,
AllocDiag, AllocDiag,
...@@ -3524,6 +3525,19 @@ class TestGetScalarConstantValue: ...@@ -3524,6 +3525,19 @@ class TestGetScalarConstantValue:
e = extract_constant(c) e = extract_constant(c)
assert e == 3, (c, d, e) 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(): def test_complex_mod_failure():
# Make sure % fails on complex numbers. # Make sure % fails on complex numbers.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论