def test_json(self):
with tempfile.TemporaryDirectory() as tempdir, self.assertWarnsRegex(
DeprecationWarning, r"Compiling to JSON is deprecated"):
temp_filepath = os.path.join(tempdir, 'output.json')
ir_utils._write_ir_to_file(json_dict, temp_filepath)
actual = load_from_file(temp_filepath)
self.assertEqual(actual, json_dict)
def save_to_component_yaml(self, output_file: str) -> None:
"""Saves ComponentSpec into yaml file.
Args:
output_file: File path to store the component yaml.
"""
ir_utils._write_ir_to_file(self.dict(), output_file)
def save_to_component_yaml(self, output_file: str) -> None:
"""Saves ComponentSpec into YAML file.
Args:
output_file: File path to store the component yaml.
"""
ir_utils._write_ir_to_file(self.to_dict(by_alias=True), output_file)
def _write_pipeline_spec_file(
self,
pipeline_spec: pipeline_spec_pb2.PipelineSpec,
package_path: str,
) -> None:
"""Writes pipeline spec into a YAML or JSON (deprecated) file.
Args:
pipeline_spec: IR pipeline spec.
package_path: The file path to be written.
"""
json_dict = json_format.MessageToDict(pipeline_spec)
ir_utils._write_ir_to_file(json_dict, package_path)
def test_json(self):
with tempfile.TemporaryDirectory() as tempdir, self.assertWarnsRegex(
DeprecationWarning, r'Compiling to JSON is deprecated'):
temp_filepath = os.path.join(tempdir, 'output.json')
ir_utils._write_ir_to_file(json_dict, temp_filepath)
with open(temp_filepath) as f:
actual_contents = f.read()
actual_dict = load_from_file(temp_filepath)
expected_contents = textwrap.dedent("""\
{
"key": "val",
"list": [
"1",
2,
3.0
]
}""")
self.assertEqual(actual_contents, expected_contents)
self.assertEqual(actual_dict, json_dict)
def test_incorrect_extension(self):
with tempfile.TemporaryDirectory() as tempdir, self.assertRaisesRegex(
ValueError, r'should end with "\.yaml"\.'):
temp_filepath = os.path.join(tempdir, 'output.txt')
ir_utils._write_ir_to_file(json_dict, temp_filepath)
def test_yml(self):
with tempfile.TemporaryDirectory() as tempdir:
temp_filepath = os.path.join(tempdir, 'output.yml')
ir_utils._write_ir_to_file(json_dict, temp_filepath)
actual = load_from_file(temp_filepath)
self.assertEqual(actual, json_dict)