def test_schema_infer(dataframe):
expect = {
'type':
'record',
'name':
'Root',
'fields': [
{
'type': ['null', 'boolean'],
'name': 'Boolean'
},
{
'type':
['null', {
'logicalType': 'timestamp-micros',
'type': 'long'
}],
'name':
'DateTime64'
},
{
'type': ['null', 'double'],
'name': 'Float64'
},
{
'type': ['null', 'long'],
'name': 'Int64'
},
{
'type': ['null', 'string'],
'name': 'String'
},
]
}
assert expect == pdx.__schema_infer(dataframe)
def test_schema_infer(dataframe):
expect = {
'type':
'record',
'name':
'Root',
'fields': [{
'type': ['null', 'boolean'],
'name': 'Boolean'
}, {
'type': ['null', 'double'],
'name': 'Float64'
}, {
'type': ['null', 'long'],
'name': 'Int64'
}, {
'type': ['null', 'string'],
'name': 'String'
}]
}
assert expect == pdx.__schema_infer(dataframe)