Exemplos de JsonSchemaParser.get_data_type em Python

Exemplo n.º 1

def test_get_data_type_array(schema_types, result_types):
    parser = JsonSchemaParser('')
    assert parser.get_data_type(JsonSchemaObject(type=schema_types)) == DataType(
        data_types=[DataType(type=r,) for r in result_types],
        is_optional='null' in schema_types,
        imports=[IMPORT_OPTIONAL] if 'null' in schema_types else [],
    )

Exemplo n.º 2

Arquivo: test_jsonschema.py Projeto: koxudaxi/datamodel-code-generator

def test_get_data_type_array(schema_types, result_types):
    parser = JsonSchemaParser('')
    assert parser.get_data_type(
        JsonSchemaObject(type=schema_types)) == parser.data_type(
            data_types=[parser.data_type(type=r, ) for r in result_types],
            is_optional='null' in schema_types,
        )

Exemplo n.º 3

Arquivo: test_jsonschema.py Projeto: languitar/datamodel-code-generator

def test_get_data_type_array(schema_types, result_types):
    parser = JsonSchemaParser(BaseModel, CustomRootType)
    assert parser.get_data_type(JsonSchemaObject(type=schema_types)) == [
        DataType(
            type=r,
            optional='null' in schema_types,
            imports_=[IMPORT_OPTIONAL] if 'null' in schema_types else None,
        ) for r in result_types
    ]

Exemplo n.º 4

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 = []

    parser = JsonSchemaParser('')
    assert parser.get_data_type(
        JsonSchemaObject(type=schema_type, format=schema_format)
    ) == DataType(type=result_type, imports=imports)

Exemplo n.º 5

Arquivo: test_jsonschema.py Projeto: shnups/datamodel-code-generator

def test_get_data_type(schema_type, schema_format, result_type, from_,
                       import_):
    if from_ and import_:
        import_: Optional[Import] = Import(from_=from_, import_=import_)
    else:
        import_ = None

    parser = JsonSchemaParser('')
    assert (parser.get_data_type(
        JsonSchemaObject(type=schema_type,
                         format=schema_format)).dict() == DataType(
                             type=result_type, import_=import_).dict())

Exemplo n.º 6

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 = JsonSchemaParser(BaseModel, CustomRootType)
    assert parser.get_data_type(
        JsonSchemaObject(type=schema_type, format=schema_format)) == [
            DataType(type=result_type, imports_=imports_)
        ]

Exemplo n.º 7

def test_get_data_type_invalid_obj():
    with pytest.raises(ValueError, match='invalid schema object'):
        parser = JsonSchemaParser(BaseModel, CustomRootType)
        assert parser.get_data_type(JsonSchemaObject())

Exemplo n.º 8

def test_get_data_type_array(schema_types, result_types):
    parser = JsonSchemaParser(BaseModel, CustomRootType)
    assert parser.get_data_type(JsonSchemaObject(type=schema_types)) == [
        DataType(type=r) for r in result_types
    ]