def test_to_yaml():
"""Test that to_yaml writes to yaml string."""
schema = _create_schema()
yaml_str = io.to_yaml(schema)
assert yaml_str.strip() == YAML_SCHEMA.strip()
yaml_str_schema_method = schema.to_yaml()
assert yaml_str_schema_method.strip() == YAML_SCHEMA.strip()
def test_to_yaml():
"""Test that to_yaml writes to yaml string."""
schema = _create_schema()
yaml_str = io.to_yaml(schema)
with tempfile.NamedTemporaryFile("w+") as f:
f.write(yaml_str)
with tempfile.NamedTemporaryFile("w+") as f:
f.write(YAML_SCHEMA)
assert yaml_str.strip() == YAML_SCHEMA.strip()
yaml_str_schema_method = schema.to_yaml()
assert yaml_str_schema_method.strip() == YAML_SCHEMA.strip()
def test_io_yaml():
"""Test read and write operation on file names."""
schema = _create_schema()
# pass in a file name
with tempfile.NamedTemporaryFile("w+") as f:
output = io.to_yaml(schema, f.name)
assert output is None
schema_from_yaml = io.from_yaml(f.name)
assert schema_from_yaml == schema
# pass in a Path object
with tempfile.NamedTemporaryFile("w+") as f:
output = schema.to_yaml(Path(f.name))
assert output is None
schema_from_yaml = pa.DataFrameSchema.from_yaml(Path(f.name))
assert schema_from_yaml == schema