import pytest
from argschema import ArgSchemaParser, ArgSchema
from argschema.fields import OptionList
import marshmallow as mm
[docs]class OptionSchema(ArgSchema):
a = OptionList([1, 2, 3], required=True, description='one of 1,2,3')
[docs]def test_option_list():
input_data = {
'a': 1
}
ArgSchemaParser(
input_data=input_data, schema_type=OptionSchema, args=[])
[docs]def test_bad_option():
input_data = {
'a': 4
}
with pytest.raises(mm.ValidationError):
ArgSchemaParser(
input_data=input_data, schema_type=OptionSchema, args=[])