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