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,
        )
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,
            imports=[IMPORT_OPTIONAL] if 'null' in schema_types else [],
        )