code: def test_file_format_values(tmp_path: pathlib.Path):
      # Default file format
      builder = testing.DummyDataset(
          data_dir=tmp_path, file_format=None
      )
      assert (
          builder.info.file_format == file_adapters.FileFormat.TFRECORD
      )
  
      # str accepted
      builder = testing.DummyDataset(
          data_dir=tmp_path, file_format="riegeli"
      )
      assert (
          builder.info.file_format == file_adapters.FileFormat.RIEGELI
      )
  
      # file_adapters.FileFormat accepted
      builder = testing.DummyDataset(
          data_dir=tmp_path,
          file_format=file_adapters.FileFormat.RIEGELI,
      )
      assert (
          builder.info.file_format == file_adapters.FileFormat.RIEGELI
      )
  
      # Unknown value
      with pytest.raises(ValueError, match="is not a valid FileFormat"):
          testing.DummyDataset(data_dir=tmp_path, file_format="arrow")
  
imports: import json
  import os
  import pathlib
  import tempfile
  import time
  from typing import Union
  import numpy
  import pytest
  import tensorflow
  from tensorflow_datasets import testing
  from tensorflow_datasets.core import dataset_builder
  from tensorflow_datasets.core import dataset_info
  from tensorflow_datasets.core import features
  from tensorflow_datasets.core import file_adapters
  from tensorflow_datasets.core import naming
  from tensorflow_datasets.core import read_only_builder
  from tensorflow_datasets.core import splits
  from tensorflow_datasets.core import utils
  from tensorflow_datasets.core.proto import dataset_info_pb2
  from tensorflow_datasets.core.proto import feature_pb2
  from tensorflow_datasets.image_classification import mnist
  from tensorflow_datasets.testing.dummy_config_based_datasets.dummy_ds_1 import dummy_ds_1_dataset_builder
  from google.protobuf import text_format
label_desc: Use image node_id 2 for calling a global function with example usage: test_file_format_values(tmp_path) without return types