Пример #1
0
def test_convert_string_space():
    space = StringSpace(length_range=Int64Range(min=1, max=2))
    converted_space = py_converters.convert_sequence_space(space)
    assert isinstance(converted_space, Sequence)
    assert converted_space.dtype == str
    assert converted_space.size_range[0] == 1
    assert converted_space.size_range[1] == 2
Пример #2
0
def test_convert_bytes_sequence_space():
    seq = BytesSequenceSpace(length_range=Int64Range(min=1, max=2))
    converted_seq = py_converters.convert_sequence_space(seq)
    assert isinstance(converted_seq, Sequence)
    assert converted_seq.dtype == bytes
    assert converted_seq.size_range[0] == 1
    assert converted_seq.size_range[1] == 2
Пример #3
0
def test_convert_double_sequence_space():
    seq = DoubleSequenceSpace(
        length_range=Int64Range(min=1, max=2), scalar_range=DoubleRange(min=3.1, max=4)
    )
    converted_seq = py_converters.convert_sequence_space(seq)
    assert isinstance(converted_seq, Sequence)
    assert converted_seq.dtype == float
    assert converted_seq.size_range[0] == 1
    assert converted_seq.size_range[1] == 2
    assert isinstance(converted_seq.scalar_range, Scalar)
    assert converted_seq.scalar_range.min == 3.1
    assert converted_seq.scalar_range.max == 4
Пример #4
0
def test_convert_float_sequence_space():
    seq = FloatSequenceSpace(
        length_range=Int64Range(min=1, max=2), scalar_range=FloatRange(min=3.1, max=4)
    )
    converted_seq = py_converters.convert_sequence_space(seq)
    assert isinstance(converted_seq, Sequence)
    assert converted_seq.dtype == np.float32
    assert converted_seq.size_range[0] == 1
    assert converted_seq.size_range[1] == 2
    assert isinstance(converted_seq.scalar_range, Scalar)
    assert np.isclose(converted_seq.scalar_range.min, 3.1)
    assert converted_seq.scalar_range.max == 4
Пример #5
0
def test_convert_int64_sequence_space():
    seq = Int64SequenceSpace(
        length_range=Int64Range(min=1, max=2), scalar_range=Int64Range(min=3, max=4)
    )
    converted_seq = py_converters.convert_sequence_space(seq)
    assert isinstance(converted_seq, Sequence)
    assert converted_seq.dtype == np.int64
    assert converted_seq.size_range[0] == 1
    assert converted_seq.size_range[1] == 2
    assert isinstance(converted_seq.scalar_range, Scalar)
    assert converted_seq.scalar_range.min == 3
    assert converted_seq.scalar_range.max == 4
Пример #6
0
def test_convert_boolean_sequence_space():
    seq = BooleanSequenceSpace(
        length_range=Int64Range(min=1, max=2),
        scalar_range=BooleanRange(min=True, max=False),
    )
    converted_seq = py_converters.convert_sequence_space(seq)
    assert isinstance(converted_seq, Sequence)
    assert converted_seq.dtype == bool
    assert converted_seq.size_range[0] == 1
    assert converted_seq.size_range[1] == 2
    assert isinstance(converted_seq.scalar_range, Scalar)
    assert converted_seq.scalar_range.min == True  # noqa: E712
    assert converted_seq.scalar_range.max == False  # noqa: E712