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
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
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
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
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
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