def setUp(self):
super(ReaderTest, self).setUp()
with absltest.mock.patch.object(example_parser, 'ExampleParser',
testing.DummyParser):
self.reader = tfrecords_reader.Reader(self.tmp_dir, 'some_spec')
self.reader.read = functools.partial(
self.reader.read,
read_config=read_config_lib.ReadConfig(),
shuffle_files=False,
)
def _tfrecords_reader(self):
return tfrecords_reader.Reader(self._data_dir, self._example_specs)
def setUp(self):
super(ReaderTest, self).setUp()
with absltest.mock.patch.object(example_parser, 'ExampleParser',
testing.DummyParser):
self.reader = tfrecords_reader.Reader(self.tmp_dir, 'some_spec')