提交 da471a8f authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者: Ricardo Vieira

Do not change global config flag in printing test

Also make test slightly more restrictive
上级 e3fb4985
...@@ -9,6 +9,7 @@ import numpy as np ...@@ -9,6 +9,7 @@ import numpy as np
import pytest import pytest
import pytensor import pytensor
from pytensor import config
from pytensor.compile.mode import get_mode from pytensor.compile.mode import get_mode
from pytensor.compile.ops import deep_copy_op from pytensor.compile.ops import deep_copy_op
from pytensor.printing import ( from pytensor.printing import (
...@@ -464,8 +465,10 @@ def test_Print(capsys): ...@@ -464,8 +465,10 @@ def test_Print(capsys):
assert "hello" in stdout assert "hello" in stdout
def test_summary(): def test_summary_with_profile_optimizer():
old_profile_optimizer_config_value = pytensor.config.profile_optimizer = True with config.change_flags(profile_optimizer=True):
f = pytensor.function(inputs=[], outputs=[], profile=True) f = pytensor.function(inputs=[], outputs=[], profile=True)
f.profile.summary()
pytensor.config.profile_optimizer = old_profile_optimizer_config_value s = StringIO()
f.profile.summary(file=s)
assert "Rewriter Profile" in s.getvalue()
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论