def test_get_data_type(schema_type, schema_format, result_type, from_, import_):
if from_ and import_:
imports_: Optional[List[Import]] = [Import(from_=from_, import_=import_)]
else:
imports_ = None
parser = OpenAPIParser(BaseModel, CustomRootType)
assert parser.get_data_type(
JsonSchemaObject(type=schema_type, format=schema_format)
) == DataType(type=result_type, imports_=imports_)
def test_get_data_type_array(schema_types, result_types):
parser = OpenAPIParser(BaseModel, CustomRootType)
assert parser.get_data_type(JsonSchemaObject(type=schema_types)) == [
DataType(type=r) for r in result_types
]