假设策略生成inf时,特别要求不要

from functools import partial
import hypothesis as h
import hypothesis.strategies as hs
import hypothesis.extra.numpy as hnp
import numpy as np


floats_notnull = partial(hs.floats, allow_nan=False, allow_infinity=False)
complex_notnull = partial(hs.complex_numbers, allow_nan=False, allow_infinity=False)

data_strategy_real = hnp.arrays(
    np.float64,
    hs.tuples(hs.integers(min_value=2, max_value=50),
              hs.integers(min_value=2, max_value=5)),
    floats_notnull()
)

data_strategy_complex = hnp.arrays(
    np.complex64,
    hs.tuples(hs.integers(min_value=2, max_value=50), hs.just(1)),
    complex_notnull()
)

data_strategy = hs.one_of(data_strategy_real, data_strategy_complex)

如果你跑了 data_strategy.example() 有几次,您会注意到结果中的一些值有无限的实部或虚部。我在这里的意图是明确地禁止无限部分或NaN部分。

我做错什么了?

更新 :如果我使用

data_strategy = hs.lists(complex_notnull, min_size=2, max_size=50)

如果我使用

data_strategy = data_strategy_real

不 inf 有人出现了。