tfds.testing.MockFs

View source on GitHub

This util wraps mock for the tf.io.gfile API.

tfds.testing.MockFs()

Usage:

fs = MockFs()
with fs.mock():

  fs.add_file('/path/to/file1', 'Content of file 1')

  assert tf.io.gfile.exists('/path/to/file1')
  with tf.io.gfile.GFile('/path/to/file2', 'w') as f:
    f.write('Content of file 2')
  tf.io.gfile.rename('/path/to/file1', '/path/to/file1_moved')

  assert fs.files == {
      '/path/to/file2': 'Content of file 2',
      '/path/to/file1_moved': 'Content of file 1',
  }

Attributes
files	Dict[str, str], mapping existing files -> file content

Methods

add_file

View source

add_file(
    path, content=None
) -> None

contextmanager

View source

@contextlib.contextmanager
contextmanager() -> Iterator['MockFs']

Open the file.

mock

View source

mock()

__enter__

View source

__enter__()

__exit__

View source

__exit__(
    exc_type, exc_value, traceback
)